最小生成树

HDU-1233

还是畅通工程

Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 50021    Accepted Submission(s): 22841

Problem Description

某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。

 

Input

测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 )；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。
当N为0时，输入结束，该用例不被处理。

 

Output

对每个测试用例，在1行里输出最小的公路总长度。

 

Sample Input

  
  

   
   3
1 2 1
1 3 2
2 3 4
4
1 2 1
1 3 4
1 4 1
2 3 3
2 4 2
3 4 5
0
  
  

 

Sample Output

  
  

   
   3
5

  
  

#include<iostream>
using namespace std;
#define INF 0x3f3f3f
struct 
{
	int v;
	int cost;
}p[105];
int a[105][105];
int main()
{
	int n,k;
	while(~scanf("%d",&n)&&n)
	{
		int i,j,c,e,f,s,mincost,ans=0;
		for(i=1;i<=n;i++)
			for(j=1;j<=n;j++)
			{
				a[i][j]=INF;
				if(i==j)
				a[i][j]=0;
			}
			k=(n*(n-1)/2);
			for(i=0;i<k;i++)
			{
				cin>>e>>f>>c;
				a[f][e]=a[e][f]=c;
			}
			s=1;
			for(i=1;i<=n;i++)
			{
				p[i].v=s;
				p[i].cost=a[s][i];
			}
			p[s].v=1;p[s].cost=0;
			for(i=1;i<n;i++)
			{
				for(j=1;j<=n;j++)
				if(p[j].cost!=0)
				{k=j;break;}			
				mincost=p[k].cost;

				for(j=1;j<=n;j++)
				if(p[j].cost!=0&&p[j].cost<mincost)
				{mincost=p[j].cost,k=j;}
				
				ans+=p[k].cost;
				p[k].cost=0;
			
				for(j=1;j<=n;j++)
				if(a[k][j]<p[j].cost)
				p[j].v=k,p[j].cost=a[k][j];
		}
		cout<<ans<<endl;
	}
}