POJ1258 Agri-Net

考查点：最小生成树

提交情况：1次TLE 2次WA

收获：当有多组数据而题目中没有说什么情况下退出时，用while(scanf()!=EOF)判断，否则TLE~

AC_CODE：

1. #include <stdio.h>
3. bool visited[101];
4. int lowcost[101],mincost,roads[101][101],min;
5. int prim(int n)
6. {
7.     int min_v;
8.     visited[0]=true;
9.     mincost=0;//开始忘了，导致2次WA~粗心呐~
10.     for(int i=0;i<n;i++)
11.         lowcost[i]=roads[0][i];
12.     for(int i=1;i<n;i++)
13.     {
14.         min=1000000;
15.         for(int j=0;j<n;j++)
16.             if(visited[j]==false&&lowcost[j]<min)
17.             {
18.                 min=lowcost[j];
19.                 min_v=j;
20.             }
21.         visited[min_v]=true;
22.         mincost+=min;
23.         for(int j=0;j<n;j++)
24.             if(visited[j]==false&&roads[min_v][j]<lowcost[j])
25.                 lowcost[j]=roads[min_v][j];
26.     }
27.     return mincost;
28. }
29. int main()
30. {
31.     int n;
32.     while (scanf("%d",&n)!=EOF)
33.     {
34.         if(n==0) break;
35.         for(int i=0;i<n;i++)
36.             visited[i]=false;
37.         for(int i=0;i<n;i++)
38.             for(int j=0;j<n;j++)
39.                 scanf("%d",&roads[i][j]);
40.         printf("%d/n",prim(n));
41.     }
42.     return 0;
43. }