pku 2485

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[124755]; 
11. int parent[505];  
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 max=-1; 
54.     int i; 
55.     int j=0; 
56.     qsort(e+1,M,sizeof(edge),cmp);  
57.     for(i=1;i<=M;i++) 
58.     {  
59.         int p1=Find(e[i].u); 
60.         int p2=Find(e[i].v); 
61.         if(p1!=p2) 
62.         { 
63.             Union(p1,p2); 
64.             if(e[i].w>max) 
65.                 max=e[i].w; 
66.             j++; 
67.         } 
68.         if(j>=N-1) 
69.             break; 
70.     } 
71.     printf("%d/n",max);  
72. }  
74. int main()  
75. {  
76.     int T; 
77.     int N,Q;  
78.     int p1,p2; 
79.     int count=1; 
80.     int i,j;  
81.     int matrix[505][505]; 
83.     scanf("%d",&T); 
84.     while(count<=T) 
85.     { 
86.         scanf("%d",&N); 
87.         memset(parent,-1,sizeof(int)*(N+1));  
88.         for(i=1;i<=N;i++) 
89.             for(j=1;j<=N;j++) 
90.                 scanf("%d",&matrix[i][j]); 
92.         int k=1; 
93.         for(i=1;i<=N;i++) 
94.             for(j=i+1;j<=N;j++)  
95.             { 
96.                 e[k].u=i; 
97.                 e[k].v=j; 
98.                 e[k].w=matrix[i][j]; 
99.                 k++; 
100.             } 
102.         KruskalMinTree(N,N*(N-1)/2);  
103.         count++; 
104.     } 
105.     return 0;  
106. }  
107. //knowledge: the minimum cost spanning tree  
108. //Submit:提交两次。1次:AC 
109. //              2次:试了一下改进的代码，结果并无效率改进  
110. //Learned:进一步了解Kruskal算法 
111. //下面是prim:
112. /*
113. #include <stdio.h> 
114. #include <string.h> 
115. #include <algorithm>
116. #define MAX 505
117. #define INF 65538
118. void prim(int matrix[][MAX],int N,int u)
119. {
120.     int i,j,k;
121.     int min;
122.     int maxcost=0;
123.     int closest[MAX];
124.     int lowcost[MAX];
126.     for(i=1;i<=N;i++)
127.     {
128.         lowcost[i]=matrix[u][i];
129.         closest[i]=u;
130.     }
132.     for(i=2;i<=N;i++)
133.     {
134.         min=INF;
135.         for(j=1;j<=N;j++)
136.             if(lowcost[j] && lowcost[j]<min)
137.             {
138.                 min=lowcost[j];
139.                 k=j;
140.             }
141.             if(maxcost<min)
142.                 maxcost=min;
143.         lowcost[k]=0;
144.         for(j=1;j<=N;j++)
145.         {
146.             if(lowcost[j] && lowcost[j]>matrix[k][j])
147.             {
148.                 lowcost[j]=matrix[k][j];
149.                 closest[j]=k;
150.             }
151.         }
152.     }
153.     printf("%d/n",maxcost);
154. }
155.         
156. int main()
157. { 
158.     int T,N; 
159.     int count=1;
160.     int i,j;
161.     int matrix[MAX][MAX];
163.     scanf("%d",&T);
164.     while(count<=T)
165.     {
166.         scanf("%d",&N);
167.         for(i=1;i<=N;i++)
168.             for(j=1;j<=N;j++)
169.                 scanf("%d",&matrix[i][j]);
170.         prim(matrix,N,1);
171.         count++;
172.     }
173.     return 0;
174. }
175. */