pku 2421

1. #include <stdio.h>
2. #include <string.h>
3. #include <algorithm>
4. typedef struct
5. {
6.     int u;
7.     int v;
8.     int w;
9. }edge;
10. edge e[5055];
11. int parent[105];
12. int Find(int e)
13. {
14.     int eP=e;
15.     while(parent[eP]>=0)
16.         eP=parent[eP];
17.     
18.     int temp;
19.     while(e!=eP)
20.     {
21.         temp=parent[e];
22.         parent[e]=eP;
23.         e=temp;
24.     }
25.     return eP;
26. }
28. void Union(int i,int j)
29. {
30.     int p1=Find(i);
31.     int p2=Find(j);
32.     if(p1==p2)
33.         return;
34.     if(parent[p1]<parent[p2])
35.     {
36.         parent[p1]+=parent[p2];
37.         parent[p2]=p1;
38.     }
39.     else
40.     {
41.         parent[p2]+=parent[p1];
42.         parent[p1]=p2;
43.     }
44. }
46. int cmp(const void *p1,const void *p2)
47. {
48.     return ((edge *)p1)->w - ((edge *)p2)->w;
49. }
51. void KruskalMinTree(int N,int M)
52. {
53.     int sum=0;
54.     int i;
55.     qsort(e+1,M,sizeof(edge),cmp);
56.     for(i=1;i<=M;i++)
57.     {
58.         int p1=Find(e[i].u);
59.         int p2=Find(e[i].v);
60.         if(p1!=p2)
61.         {
62.             Union(p1,p2);
63.             sum+=e[i].w;
64.         }
65.     }
66.     printf("%d/n",sum);
67. }
69. int main()
70. {
71.     int N,Q;
72.     int p1,p2;
73.     int i,j,k=1;
74.     int matrix[105][105];
76.     while(scanf("%d",&N)!=EOF)
77.     {
78.         memset(parent,-1,sizeof(int)*(N+1));
79.         for(i=1;i<=N;i++)
80.             for(j=1;j<=N;j++)
81.                 scanf("%d",&matrix[i][j]);
82.         for(i=1;i<=N;i++)
83.             for(j=i+1;j<=N;j++)
84.             {
85.                 e[k].u=i;
86.                 e[k].v=j;
87.                 e[k].w=matrix[i][j];
88.                 k++;
89.             }
90.         scanf("%d",&Q);
91.         for(i=1;i<=Q;i++)
92.         {
93.             scanf("%d%d",&p1,&p2);
94.             Union(p1,p2);
95.         }
96.         KruskalMinTree(N,N*(N-1)/2);
97.     }
98.     return 0;
99. }
100. //knowledge: the minimum cost tree
101. //Submit:提交两次。1次WA因为没有考虑一次多组输入和没读懂题意：求所有路径的sum值
102. //           2次AC
103. //Learned:进一步了解了Kruskal算法