HDU1233 还是畅通工程

 1 #include <iostream>
 2 #include <algorithm>
 3 #define MAXN 99*50
 4 using namespace std;
 5 
 6 int pre[105];　　//存储当前节点的前驱结点
 7 int n, m;
 8     
 9 struct node{　　　　//结构体存储边
10     int begin;
11     int end;
12     int len; 
13 }s[MAXN];
14 
15 bool cmp(node a, node b)
16 {
17     return a.len < b.len;　　//边小的排前面
18 }
19 
20 void init(int n)　　//初始化成独立节点
21 {
22     for(int i = 1; i <= n; i++)
23         pre[i] = i;
24 }
25 
26 int find(int x)　　//寻找当前结点的根节点
27 {
28     while(x != pre[x])
29         x = pre[x];
30     
31     return x;
32 }
33 
34 void merge(int fx, int fy)　　//归并两个不同根节点的结点
35 {
36     if(fx != fy)
37         pre[fx] = fy;
38 }
39 
40 int kruskal()
41 {
42     int minlen = 0;
43     sort(s, s + m, cmp);　　//让边从小到大排序
44     for(int i = 0; i < m; i++)
45     {
46         int fx = find(s[i].begin);
47         int fy = find(s[i].end);
48         if(fx != fy)
49         {
50             merge(fx, fy);
51             minlen += s[i].len;　　//归并后求最小路径和
52         }
53     } 
54     return minlen;
55  } 
56 
57 int main()
58 {
59     while(scanf("%d", &n) != EOF && n)
60     {
61         init(n);
62         m = n * (n - 1) / 2;
63         for(int i = 0; i < m; i++)
64         {
65             scanf("%d%d%d", &s[i].begin, &s[i].end, &s[i].len);
66         }
67         printf("%d\n", kruskal());
68     }
69     return 0;
70 }