poj2421修建道路

题目

有N个村庄，编号从1到N，你应该建造一些道路，这样每两个村庄就可以相互连接。我们说两个村庄A和B是相连的，当且仅当A和B之间有一条路，或者存在一个村庄C，使得A和C之间有一条路，C和B是相连的。
我们知道，一些村庄之间已经有一些道路，你的工作是修建一些道路，使所有村庄都连接起来，并且所有道路的长度都是
最小的。

输入

第一行是整数 N （3 <= N <= 100），这是村庄的数量。然后是N条线，其中第i条包含N个整数，这些N个整数的j是村庄i和村庄j

之间的距离 < <（距离应该是[1，1000]内的一个整数）。然后是Q线，每条线包含两个整数a和b（1<= a<b<= N），这意味着村庄a和村庄b之间的道路已经建成。

输出

您应该输出一条包含整数的线，该整数是要建造的所有道路的长度，以便所有村庄都连接起来，并且此值是最小值。

示例输入

3
0 990 692
990 0 179
692 179 0
1
1 2

示例输出

179

设计思路

将已经修建好的路的权值置为0，这样求和就不用重复计算，保证了这些路的优先选择。先把所有的道路从小到大排序，然后从小到大依次考察每条边，如果这条边的两个端点不属于同一集合，那么就将它们合并，直到所有的点都属于同一个集合为止。这里用到了并查集：查询元素a和元素b是否为同一组，将元素a和b合并为同一组。

算法流程：

输入： 图G
输出： 图G的最小生成树
具体流程：
(1)将图G看做一个森林，每个顶点为一棵独立的树
(2)将所有的边加入集合S，即一开始S = E
(3)从S中拿出一条最短的边(u,v)，如果(u,v)不在同一棵树内，则连接u,v合并这两棵树，同时将(u,v)加入生成树的边集E'
(4)重复(3)直到所有点属于同一棵树，边集E'就是一棵最小生成树

如下图最小生成树：

 

代码如下

#include<iostream>
#include<algorithm>
using namespace std;
int t[100];
struct edge {
	int r, l, w;
}e[10000];
int map[100][100];
int flag[100][100];
int nedge = 0;
bool cmp(edge x, edge y) {
	return x.w < y.w;
}
int find(int x) {
	if (t[x] == x)
		return x;
	else
		return find(t[x]);
}
void Union(int x, int y) {
	int xt = t[x];
	int yt = t[y];
	if (xt != yt) {
		t[xt] = yt;
	}
}
int kruskal() {
	int t1, t2, ans = 0;
	for (int i = 0; i < nedge; i++) {
		t1 = find(e[i].l);
		t2 = find(e[i].r);
		if (t1 != t2) {
			ans += e[i].w;
			Union(t1, t2);
		}
	}
	return ans;
}
int main()
{
	int n,i,j,m;
	while (~scanf("%d", &n)&&n) {
		for (int i = 1; i <= n; i++) {
			t[i] = i;
		}
		for (i = 1; i <= n; i++) {
			for (j = 1; j <= n; j++) {
				cin>>map[i][j];
			}
		}
		for (i = 1; i < n; i++) {
			for (j = i + 1; j <= n; j++) {
				e[nedge].l = i;
				e[nedge].r = j;
				e[nedge].w = map[i][j];
				nedge++;
			}
		}
		cin>>m;
		while (m--) {
			int a, b;
			cin>>a>>b;
			int t1 = find(a);
			int t2 = find(b);
			Union(t1, t2);
		}
		sort(e, e + nedge, cmp);
		cout << kruskal() << endl;
	}
	return 0;
}

运行结果：

 

大数据201 tyx