POJ 2485 Highways

这是prim的一个变型,求最小生成树中，最大的一条边,代码写法基本上也是一样的

1. #include<iostream>
2. #include<cstdio>
3. #include <vector>
4. using namespace std;
5. const int MAX = 501;
6. const int INF = 65537;
7. int c[MAX][MAX];
8. bool flag[MAX];
9. vector<int> U_Set;
10. //这道题目是prim的一个变形,求最小生成树中的，最大的一段长度
11. int Prim(int n)
12. {
13.         int ans = 0;
14.         U_Set.push_back(1);
15.         flag[1] = true;
16.         for(int i = 1; i < n; i++)
17.         {
18.             int start = 1;
19.             int k = 1;
20.             int end = 1;
21.             int min = INF;
22.             int U_len = U_Set.size();
23.             for(int p=0;p<U_len;p++) {
24.                 start = U_Set[p];//每次取出U集合中一个点,与不是U集合中的点比较,取路程最小的一个
25.                 for(k=1;k<=n;k++)
26.                 {
27.                     if(start!=k&&c[start][k]<min&&!flag[k])
28.                     {
29.                         min = c[start][k];
30.                         end = k;
31.                     }
32.                 }
33.             }
34.             //找到那个点以后,flag置为true
35.             flag[end] = true;
36.             U_Set.push_back(end);
37.             //边为start----end
39.             //找到最小生成树中最大的那段
40.             if(ans<min)
41.                ans = min;
42.         }
43.         return ans;
44. }
45. int main() {
46.     int i, j, n, m,T=1;
47.     cin>>T;
48.     while(T-->0) {
49.         cin>>n;
50.         for(i = 1; i <= n; i++)
51.         {
52.             for(j = 1; j <=n; j++)
53.             {
54.                 //这里用cin会TLE
55.                 scanf("%d",&c[i][j]);
56.             }
57.         }
58.     cout<<Prim(n)<<endl;
59.     memset(flag,0,sizeof(flag));
60.     U_Set.clear();
61.     }
62.     return 0;
63. }