Constructing Roads POJ - 2421 (最小生成树)

思路：首先使用二维数组dis[][]处理输入, 对于已经修好的路，将其对应的dis[i][j]置为零即可。最后再将

　　    所有的dis[][]保存到边结构体中，使用Kruskal算法求得最小生成树。

 1 #include<iostream>
 2 #include<vector>
 3 #include<string>
 4 #include<cmath>
 5 #include<set>
 6 #include<algorithm>
 7 #include<cstdio>
 8 #include<map>
 9 #include<cstring>
10 
11 using namespace std;
12 
13 int dis[110][110];
14 struct Edge
15 {
16     int a, b;
17     int cost;
18 }edge[1000010];
19 
20 int Tree[110];
21 
22 int findRoot(int x)
23 {
24     if(Tree[x] == -1)
25         return x;
26     int tmp = findRoot(Tree[x]);
27     Tree[x] = tmp;
28     return tmp;
29 }
30 
31 bool cmp(Edge e1, Edge e2)
32 {
33     return e1.cost < e2.cost;
34 }
35 
36 int main()
37 {
38     int n;
39     scanf("%d", &n);
40     for(int i = 1; i <= n; ++i)
41         Tree[i] = -1;
42         
43     for(int i = 1; i <= n; ++i)
44         for(int j = 1; j <= n; ++j)
45             scanf("%d", &dis[i][j]);
46     
47     int q;
48     scanf("%d", &q);
49     for(int i = 1; i <= q; ++i)
50     {
51         int a, b;
52         scanf("%d %d", &a, &b);
53         dis[a][b] = 0;
54     }
55     
56     int k = 1;
57     for(int i = 1; i <= n; ++i)
58         for(int j = 1; j <= n; ++j)
59         {
60             if(i != j && i < j)    // 注意同一条边不要重复保存！ 
61             {
62                 edge[k].a = i;
63                 edge[k].b = j;
64                 edge[k].cost = dis[i][j];
65                 ++k;
66             }
67         }
68     sort(edge+1, edge+k, cmp);
69     int ans = 0;
70     for(int i = 1; i < k; ++i)
71     {
72         int ra = findRoot(edge[i].a);
73         int rb = findRoot(edge[i].b);
74         if(ra != rb)
75         {
76             Tree[ra] = rb;
77             ans += edge[i].cost;
78         }    
79     }
80     
81     printf("%d\n", ans);
82     
83     return 0;
84 }