首页 > 代码库 > POJ2377

POJ2377

题目链接:http://poj.org/problem?id=2377

解题思路:

  Warning ! 注意考虑重边 !

  其实就是求最大生成树,没什么好说的,就上面那个坑。

AC代码:

 1 #include <iostream>
 2 #include <cstdio>
 3 using namespace std;
 4 const int maxn=1005,inf=0x7fffffff;
 5 int cost[maxn][maxn];
 6 int d[maxn];
 7 int vis[maxn];
 8 int main()
 9 {
10     int N,M;
11     scanf("%d%d",&N,&M);
12     int a,b,c;
13     for(int i=1;i<=N;i++)
14         for(int j=1;j<=i;j++){
15             if(i==j)    cost[i][j]=0;
16             else    cost[i][j]=cost[j][i]=-inf;
17         }
18     for(int i=0;i<M;i++){
19         scanf("%d%d%d",&a,&b,&c);
20         if(cost[a][b]>-inf){
21             if(cost[a][b]<c)    cost[a][b]=cost[b][a]=c;
22         }
23         else    cost[a][b]=cost[b][a]=c;
24     }
25     for(int i=1;i<=N;i++){
26         d[i]=-inf;
27         vis[i]=0;
28     }
29     d[1]=0;
30     long long res=0;
31     while(1){
32         int v=-1;
33         for(int u=1;u<=N;u++){
34             if(!vis[u]&&(v==-1||d[u]>d[v])) v=u;
35         }
36         if(v==-1)   break;
37         vis[v]=1;
38         res+=d[v];
39         for(int u=1;u<=N;u++){
40             d[u]=max(d[u],cost[v][u]);
41         }
42     }
43     if(res<0)   printf("-1\n");
44     else    printf("%lld\n",res);
45     return 0;
46 }

 

POJ2377