最小生成树&并查集练习

丛林中的路

【题目描述】：

热带岛屿Lagrishan的首领现在面临一个问题：几年前，一批外援资金被用于维护村落之间的道路，但日益繁茂的丛林无情的侵蚀着村民的道路，导致道路维修开销巨大，长老会不得不放弃部分道路的维护。已经知道了每条路每个月的维修费用（单位为aacms）。现在长老会需要提出一种方案，即需要保证村落之间都可以互相到达，又要将每个月的道路维修费用控制在最小。村子编号为从A到I。上图右侧显示的方案最小维修开销为216 aacms每月。

【输入描述】：

输入包含1~50个数据集，最后一行为0.每个数据集第一行为村落数目n, 1 < n < 27,依次用字母表的前n个字母标记。接下来有n-1行，每行的第一个数据便是按字母顺序排列的村子编号（不包括最后一个村庄）。每个村庄后面的数据k代表该村庄通往编号在其之后的村庄的道路数目，如A 2 B 12 I 25，代表A村庄有2个编号在A之后的村庄和其相连。若k大于0，k后面会依次给出这k个村庄的编号以及各自到起始村庄的道路维修费用，如A 2 B 12 I 25，代表A和B之间道路维修费用为12， A和I之间道路维修费用为25（维修费用为不超过100的正整数）.路的总数目不超过75条，每个村庄到其他村庄不会有超过15条路（包括编号在其之前和之后的）。

【输出描述】：

每个数据集有一个输出：针对解决方案每个月维修道路的小费用。

提示：蛮力算法虽能找出解决方案，但将会超出时间限制。

【样例输入】：

9
A 2 B 12 I 25
B 3 C 10 H 40 I 8
C 2 D 18 G 55
D 1 E 44
E 2 F 60 G 38
F 0
G 1 H 35
H 1 I 35
3
A 2 B 10 C 40
B 1 C 20
0

【样例输出】：

216
30

【时间限制、数据范围及描述】：

时间：1s 空间：64M

这题要用到的算法是最小生成树。其实最小生成树有两种常用的算法： Prim 和 Kruskal 算法。在这里我们只介绍Kruskal算法。

算法大致思路如下：

•对于图G，每次寻找一条边，此边满足

•1)  边的两个端点不同在mst中

•2)  此边的权值在所有“边的两个端点不同在mst中”的边里为最小值

•3)  将这条边加入mst中。

（mst为一个集合）

那么显然这种算法必须用到并查集。

读题，题目意思就是要求我们在图中建立一个最小生成树，求这个最小生成树每条边的权值之和。所以我们首先把每条边读进来（读入可能有点麻烦，这里因为每个村庄名字都是按字母序排列的，所以点A可以看成点0，点B可以看成点2，以此类推），接着按照权值大小进行一次从小到大的排序。然后我们从小到大选择最小生成树的边，只要起点和终点不在一个集合内我们就选择这条边并将起点和终点所在的集合放到同一个集合内。这样顺次做下来就是最小生成树了。

AC代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
struct tree{
	int u,v,w;
	char nm;
}t[1005];
int n,m,f[1005],k,k0;
char s0,nm;
bool cmp(tree x,tree y){
	return x.w<y.w;
}
int find(int x){
	if(f[x]==x)return f[x];
	else return f[x]=find(f[x]);
}
void unin(int x,int y){
	f[x]=y;
}
void work(){
	int ans=0,cnt=0;
	k=0;
	for(int i=0;i<=n;i++){
		f[i]=i;
	}
	for(int i=0;i<n-1;i++){
		scanf("%c %d ",&nm,&k0);
		for(int j=0;j<k0;j++){
		    t[++k].u=nm-'A';
			scanf("%c %d ",&s0,&t[k].w);
			t[k].v=s0-'A';
		}
		scanf("\n");
    }
	sort(t+1,t+1+k,cmp);
	int fx,fy;
	for(int i=1;i<=k;i++){
		fx=find(t[i].u);fy=find(t[i].v);
		if(fx!=fy){
			ans+=t[i].w;
			cnt++;
			if(cnt==n-1)break; 
		    unin(fx,fy);
		}
	}
	printf("%d\n",ans);
}
int main(){
	while(1){
		scanf("%d\n",&n);
		if(n==0)break;
		work();
	}
	return 0;
}