POJ 1258 Agri-Net

这是一个典型的Prim算法应用，在具体的这个1258中碰到一些小accident,

由于是多case的,所以每次都需要把flag,U_Set清空掉,便于下一次执行

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