EOJ-2529 强大的lwc

http://acm.cs.ecnu.edu.cn/problem.php?problemid=2529

题意,求修造为通路耗费的最小精力。

即求无向图的最小生成树,此题注意结点是0到n 为n+1个,判断连通条件可以判断树的边数是不是为n,此题用prim

 1 #include<map>
 2 #include<cmath>
 3 #include<cctype>
 4 #include<cstdio>
 5 #include<string>
 6 #include<cstdlib>
 7 #include<cstring>
 8 #include<iostream>
 9 #include<algorithm>
10 using namespace std;
11 const int inf=999999999;
12 int mat[105][105];
13 int n;
14 void init(){
15     for(int i=0;i<=n;i++)
16         for(int j=0;j<=n;j++)
17             mat[i][j]=inf;
18 }
19 int prim(){                                  //注意 ,都是从0-n
20     int lowcost[105],s[105];
21     int ans=0,cnt=0;
22     for(int i=0;i<=n;i++){
23         lowcost[i]=mat[0][i];
24         s[i]=0;
25     }
26     s[0]=1;
27     for(int i=0;i<=n;i++){
28         int min=inf,k;
29         for(int j=0;j<=n;j++)
30             if(!s[j] && lowcost[j]<min){
31                 min=lowcost[j];
32                 k=j;
33             }
34         if(min==inf) break;
35         ans+=min;
36         cnt++;
37         s[k]=1;
38         for(int j=0;j<=n;j++)
39             if(!s[j] && mat[k][j]<lowcost[j])
40                 lowcost[j]=mat[k][j];
41     }
42     if(cnt==n)           //判断是否连通
43         return ans;
44     else return 0;
45 }
46 int main(){
47     int t;
48     cin>>t;
49     while(t--){
50         int m;
51         scanf("%d%d",&n,&m);
52         init();
53         while(m--){
54             int a,b,c;
55             scanf("%d%d%d",&a,&b,&c);
56             if(c<mat[a][b])
57                 mat[a][b]=mat[b][a]=c; //注意是无向
58         }
59         /*for(int i=0;i<=n;i++){
60             for(int j=0;j<=n;j++)
61                 printf("%d ",mat[i][j]);
62             cout<<endl;
63         }*/
64         int ans=prim();
65         if(ans)printf("%d\n",ans);
66         else printf("-1\n");
67     }
68     return 0;
69 }                                                                                
View Code

 

posted on 2013-06-12 03:33  KimKyeYu  阅读(211)  评论(0编辑  收藏  举报

导航