PTA-公路村村通 (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

/*
Kruskal算法 
*/
#include<cstdio>
#include<iostream>
#include<algorithm>
#define maxn 1005
#define maxm 3005
using namespace std;

int p[maxn], u[maxm], v[maxm], w[maxm], r[maxm];

int cmp(const int i, const int j)
{
	return w[i] < w[j];
}

int find(int x)
{
	return p[x] == x ? x : p[x] = find(p[x]);
}

int main()
{
	int n, m;
	scanf("%d%d", &n, &m);
	for (int i = 1; i <= n; i++)
	{
		p[i] = i;
	}
	for (int i = 0; i < m; i++)
	{
		r[i] = i;
	}
	int ans = 0;
	for (int i = 0; i < m; i++)
	{
		scanf("%d%d%d", &u[i], &v[i], &w[i]);
	}
	sort(r, r+m, cmp);
	for (int i = 0; i < m; i++)
	{
		int e = r[i];
		int x = find(u[e]);
		int y = find(v[e]);
		if (x != y)
		{
			p[x] = y;
			ans += w[e];
		}
	}
	int flag = 1, f = find(1);
	for (int i = 2; i <= n; i++)
	{
		if (find(i) != f)
		{
			flag = 0;
			break;
		}
	}
	if (flag)
	{
		printf("%d", ans);
	}
	else
	{
		printf("-1");
	}
	return 0;
}

 

    原文作者:道路修建问题
    原文地址: https://blog.csdn.net/qq_42367703/article/details/87771951
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞