最小生成树——最短网络Agri-Net

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<algorithm>
 4 using namespace std;
 5 
 6 struct point{
 7     int x;
 8     int y;
 9     int w;
10 }a[10000];
11 int dad[101];
12 int n,m,x,k,tot;
13 
14 int find(int x)
15 {
16     if(dad[x]!=x) dad[x]=find(dad[x]);
17     return dad[x];
18 }
19 
20 void unionn(int a,int b)
21 {
22     int r1=find(a);
23     int r2=find(b);
24     if(r1!=r2)
25     dad[r1]=r2;
26 }
27 
28 int cmp(const point &a,const point &b)
29 {
30     if(a.w<b.w) return 1;
31     else return 0;
32 }
33 
34 int main()
35 {
36     cin>>n;
37     for(int i=1;i<=n;i++)
38       for(int j=1;j<=n;j++){
39           cin>>x;
40           if(x!=0){
41               m++;
42               a[m].x=i,a[m].y=j,a[m].w=x;
43           }
44       }
45     for(int i=1;i<=n;i++)
46      dad[i]=i;
47     sort(a+1,a+m+1,cmp);
48     for(int i=1;i<=m;i++){
49         if(find(a[i].x)!=find(a[i].y)){
50             unionn(a[i].x,a[i].y);
51             tot+=a[i].w;
52             k++;
53         }
54         if(k==n-1) break;
55     }
56     cout<<tot;
57     return 0;
58 }