题目链接:http://poj.org/problem?id=2377
解题思路:
Prim算法。
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 "); 44 else printf("%lld ",res); 45 return 0; 46 }