[图论]连接岛屿问题(MST.Kruskal算法 并查集)

/*
Name:连接岛屿问题 (MST Kruskal算法)

Actor:HT

Time:2015年8月13日

Error Reporte:

*/
#include 
#include 
#include 
#include 
#include 
#include 

#define N 50

using namespace std;

//最小生成树:
//依次选择可用的最小边连接,直到结束
//反证法:
//如果存在不用最小边的最佳结果,
//那么连入最小边以后,原结果成环;
//去掉可去的任意一个边,都比去掉最小边要长
//所以可证,要取最小边

struct edge			//以边为研究对象
{
	int x, y, l;
}ed[N];

int pa[N];
int s;

bool cmp(edge a, edge b)
{
	return (a.l < b.l);
}

int findpa(int a)		//并查集.递归找根
{
	if (pa[a] == a) return a;
	else return findpa(pa[a]);
}

int main()
{
	int i, j;
	int n;
	int a, b, len;
	int sum,tempa,tempb;
	while (scanf("%d %d", &s, &n) != EOF)
	{
		sum = 0;
		for (i = 0; i < n; i++)
		{
			scanf("%d %d %d", &a, &b, &len);
			ed[i].x = a;
			ed[i].y = b;
			ed[i].l = len;
		}
		for (i = 0; i < s; i++)	//并查集初始化
		{
			pa[i] = i;
		}
		sort(ed, ed+n, cmp);	//MST.边由小到大排序然后依次找
		for (i = 0; i < n; i++)
		{
			tempa = findpa(ed[i].x);
			tempb = findpa(ed[i].y);
			if (tempa != tempb)
			{
				pa[tempb] = tempa;	//并查集.标记新根  注意点,一定是把根连在一起,否则可能出现遗漏
				sum += ed[i].l;
				printf("%d-----%d     %d\n", ed[i].x, ed[i].y, ed[i].l);
			}
		}
		printf("%d\n", sum);
	}
}

你可能感兴趣的:(算法与结构)