【id:747】【20分】B. 村村通工程（Kruskal算法）

题目描述

"村村通"是国家一个系统工程，其包涵有：公路、电力、生活和饮用水、电话网、有线电视网、互联网等等。

村村通公路工程，是国家为构建和谐社会，支持新农村建设的一项重大举措，是一项民心工程。又称“五年千亿元”工程

该工程是指中国力争在5年时间实现所有村庄通沥青路或水泥路，以打破农村经济发展的交通瓶颈，解决9亿农民的出行难题。

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

要求用Kruskal算法求解

输入

第1行：顶点数n

第2行：n个顶点编号

第3行：边数m

接着m行：m条边信息，格式为：顶点1 顶点2 权值

输出

第1行：输出最小生成树的权值之和

接着n-1行对应n-1条边信息

如果能找到最小生成树，按树的生长顺序输出, 边顶点按数组序号升序输出

如果输入数据不足以保证畅通，则直接输出?1，无需输出任何边信息

输入输出样例

输入样例1 <-复制
6
v1 v2 v3 v4 v5 v6 
10
v1 v2 6
v1 v3 1
v1 v4 5
v2 v3 5
v2 v5 3
v3 v4 5
v3 v5 6
v3 v6 4
v4 v6 2
v5 v6 6

输出样例1
15
v1 v3 1
v4 v6 2
v2 v5 3
v3 v6 4
v2 v3 5

AC代码

#include<iostream>
#include<string>
#include<algorithm>
using namespace std;

string dian[20];
int p[20];
int n, m;
int weight;
struct bian
{
	string a, b;
	int weight;
	bool operator<(const bian& b2)
	{
		return weight < b2.weight;
	}
};
bian guanxi[20];
bian jieguo[20];

int xun(string a)
{
	int i;
	for (i = 0; i < n; i++)
		if (a == dian[i])
			return i;
}

int find(int x)
{
	if (x != p[x])
		p[x] = find(p[x]);
	return p[x];
}

int kruskal()
{
	sort(guanxi, guanxi + m);
	int bshu = 0;
	int i;
	weight = 0;
	int k = 0;
	for (i = 0; i < n; i++)
		p[i] = i;
	for (i = 0; i < m; i++)
	{
		int a = xun(guanxi[i].a);
		int b = xun(guanxi[i].b);
		a = find(a);
		b = find(b);
		if (a != b)
		{
			jieguo[k++] = guanxi[i];
			if (xun(guanxi[i].a) > xun(guanxi[i].b))
				swap(jieguo[k - 1].a, jieguo[k-1].b);
			p[a] = b;
			bshu++;
			weight += guanxi[i].weight;
		}
	}
	return bshu == n - 1;
}

int main()
{
	cin >> n;
	int i, j;
	for (i = 0; i < n; i++)
		cin >> dian[i];

	cin >> m;
	for (i = 0; i < m; i++)
	{
		cin >> guanxi[i].a >> guanxi[i].b >> guanxi[i].weight;
	}


	if (!kruskal())
	{
		cout << "-1" << endl;
		return 0;
	}

	cout << weight << endl;
	for (i = 0; i < n - 1; i++)
	{
		cout << jieguo[i].a << " " << jieguo[i].b << " " << jieguo[i].weight << endl;
	}
}

（by 归忆）