HDU 1233 还是畅通工程
prime()算法,计算从1到n的最小总长度
//prime算法
#include<stdio.h>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#include<math.h>
#include<set>
#include<vector>
using namespace std;
#define INF 0x3f3f3f3f
int g[200][200];
int n,m;
int dis[200],vis[200];
int prime()
{
int minn,temp;
int sum=0;
for(int i=1;i<n;i++)
{
minn=INF;
for(int j=1;j<=n;j++)
{
if(vis[j]==0 && dis[j]<minn)
{
minn=dis[j];
temp=j;
}
}
vis[temp]=1;
sum+=minn;
for(int j=1;j<=n;j++)
{
if(vis[j]==0 && dis[j]>g[temp][j])
{
dis[j]=g[temp][j];
}
}
}
return sum;
}
int main()
{
while(scanf("%d",&n)!=EOF)
{
if(n==0)
break;
m=(n*(n-1))/2;
memset(vis,0,sizeof(vis));
memset(dis,0,sizeof(dis));
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
if(i==j)
g[i][j]=0;
else
g[i][j]=INF;
}
}
for(int i=1;i<=m;i++)
{
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
g[u][v]=g[v][u]=min(g[u][v],w);
}
for(int i=1;i<=n;i++)
{
dis[i]=g[1][i];
}
dis[1]=0;
vis[1]=1;
cout<<prime()<<endl;
}
return 0;
}