acm/prime at master · zhuangAnjun/acm · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
#include<iostream>
#include<queue>
#include<cstring>
using namespace std;
int vis[500],road[500][500],q[500];
void ini()
{
    memset(vis,0,sizeof(vis));
    for(int i=0;i<500;i++)
    {
        for(int j=0;j<500;j++)
        {
            road[i][j]=99999999;
        }
    }
    q[0]=1;
}
int prime(int n)
{
    int sum=1,temp1=1,Min,ml=0,temp,place=1;
    while(sum<n)
    {
        Min=99999999;
        for(int j=0;j<place;j++)
        {
        for(int i=1;i<=n;i++)
        {
            if(road[q[j]][i]<Min&&vis[i]==0)
            {
                Min=road[q[j]][i];
                temp=i;
                temp1=q[j];
            }
        }
        }
      //  cout<<temp1<<' '<<temp<<endl;
        sum++;
        vis[temp]=1;
        ml+=Min;
        q[place++]=temp;
    }
    return ml;
}

int main()
{
    int n,m;
    while(cin>>n>>m)
    {
        ini();
        for(int i=0;i<m;i++)
        {
            int a,b,c;
            cin>>a>>b>>c;
            road[a][b]=c;
            road[b][a]=c;
        }
        vis[1]=1;
        int ml=prime(n);
        cout<<ml<<endl;
    }
}