还是畅通工程

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

Hint

Hint Huge input, scanf is recommended.

题意描述:

有n个村庄n*(n-1)/2条道路,给出了道路连接的两个村庄及权值,求出所有村庄连接在一起的最小权值和。

解题思路:
最小生成树prim算法和kruskal算法都可用,模板题。

#include<stdio.h>
#include<algorithm>
int f[110]; 
using namespace std;
struct A
{
	int a;
	int b;
	int c;
}q[5010];
int cmp(struct A x,struct A y)
{
	return x.c<y.c;
}
int getf(int i)
{
	if(f[i]==i)
		return i;
	else
	{
		f[i]=getf(f[i]);
		return f[i];
	}
}
int merge(int u,int v)
{
	int t1,t2;
	t1=getf(u);
	t2=getf(v);
	if(t1!=t2)
	{
		f[t2]=t1;
		return 1;
	}
	return 0;
}
int main()
{
	int n,m,i,sum,cut;
	while(scanf("%d",&n))
	{
		if(n==0)
			break;
		m=n*(n-1)/2;
		for(i=1;i<=m;i++)
			scanf("%d%d%d",&q[i].a,&q[i].b,&q[i].c);
		sort(q+1,q+m+1,cmp);
		for(i=1;i<=n;i++)
			f[i]=i;
		cut=0;
		sum=0;
		for(i=1;i<=m;i++)
		{
			if(merge(q[i].a,q[i].b))
			{
				cut++;
				sum=sum+q[i].c;
			}
			if(cut==n-1)
				break;
		}
		printf("%d\n",sum);
	}
	return 0;
}