7-1 公路村村通 (30 分)

现有村落间道路的统计数据表中,列出了有可能建设成标准公路的若干条道路的成本,求使每个村落都有公路连通所需要的最低成本。

输入格式:
输入数据包括城镇数目正整数N(≤1000)和候选道路数目M(≤3N);随后的M行对应M条道路,每行给出3个正整数,分别是该条道路直接连通的两个城镇的编号以及该道路改建的预算成本。为简单起见,城镇从1到N编号。

输出格式:
输出村村通需要的最低成本。如果输入数据不足以保证畅通,则输出−1,表示需要建设更多公路。

输入样例:

6 15
1 2 5
1 3 3
1 4 7
1 5 4
1 6 2
2 3 4
2 4 6
2 5 2
2 6 6
3 4 6
3 5 1
3 6 1
4 5 10
4 6 8
5 6 3

输出样例:

12

#include 
#define MAXSIZE 1001
using namespace std;
int bcj[1001];
struct rur
{
	int x;
	int y;
	int wei;
}r[3*MAXSIZE];
typedef struct rur Made;
bool cmp(Made F,Made E)
{
	return F.wei<E.wei;
}
int Found(int a);
int main()
{
	int n,m,sum=0;      //sum记录花费 
	cin>>n>>m;
	for (int i=1;i<=n;i++)    //构造群集合(并查集)
	{
		bcj[i]=i;
	}
	int a,b,wei;
	for (int i=0;i<m;i++)
	{
		cin>>a>>b>>wei;
		r[i].x=a;
		r[i].y=b;
		r[i].wei=wei;
	}
	sort(r,r+m,cmp);   //排序
	int now=0;     //已经添加的边数 
	for (int i=0;i<m;i++)      //由小到大  
	{
		int Bossx,Bossy;
		Bossx=Found(r[i].x);
		Bossy=Found(r[i].y);
		if (Bossx==Bossy)
		{
			continue;
		}
		sum+=r[i].wei;
		now++;
		if (now==n-1)
		{
			break;
		}
		bcj[Bossx]=Bossy;
	}
	if (now<n-1)    //输入的数据不足 
	{
		printf("-1\n");
//printf("**%d**\n",now);
		return 0;
	}
	printf("%d",sum);
	return 0;
}


int Found(int a)  //找a的老大是谁 
{
	if (a==bcj[a]) return a;
	else
	{
		return Found(bcj[a]);   //无压缩 
	}
}

你可能感兴趣的:(算法PTA题解)