POJ 1258 (最小生成树 kruskal)

克鲁斯卡尔（Kruskal）算法（只与边相关）

算法描述：克鲁斯卡尔算法需要对图的边进行访问，所以克鲁斯卡尔算法的时间复杂度只和边又关系，可以证明其时间复杂度为O（eloge）。

算法过程：

1.将图各边按照权值进行排序

2.将图遍历一次，找出权值最小的边，（条件：此次找出的边不能和已加入最小生成树集合的边构成环），若符合条件，则加入最小生成树的集合中。不符合条件则继续遍历图，寻找下一个最小权值的边。

3.递归重复步骤1，直到找出n-1条边为止（设图有n个结点，则最小生成树的边数应为n-1条），算法结束。得到的就是此图的最小生成树。

克鲁斯卡尔（Kruskal）算法因为只与边相关，则适合求稀疏图的最小生成树。而prime算法因为只与顶点有关，所以适合求稠密图的最小生成树。

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

struct edge{
	int u, v, w;
};

const int NODE_NUM = 102;
edge e[NODE_NUM*NODE_NUM];
int father[NODE_NUM];
int n, ne;

bool cmp(const edge& a, const edge& b)
{
	return a.w < b.w;
}

void make_set()
{
	for (int i = 1; i <= n; ++i)
		father[i] = i;
}

int find_set(int i)
{
	if (father[i] != i){
		father[i] = find_set(father[i]);
	}
	return father[i];
}

bool union_set(int a, int b) //a --> b
{
	a = find_set(a);
	b = find_set(b);
	if (a != b){  //没有共同祖先，说明没有形成回路
		father[a] = b; //将节点纳入最小生成树集合
		return true;
	}
	else{
		return false;
	}
}

int kruskal()
{
	int i, mst_edge = 0, sum = 0;
	make_set();
	sort(e, e+ne, cmp);  //将边按升序排序
	for (i = 0; i < ne; ++i){
		//如果加入的边不会使树形成回路
		if (union_set(e[i].u, e[i].v)){  
			sum += e[i].w;
			//如果纳入的边数等于顶点数-1，则说明最小生成树形成
			if (++mst_edge == n - 1){ 
				return sum;
			}
		}
	}
	return mst_edge;
}

int main()
{
	int i, j, cost;
	while (scanf("%d", &n) != EOF){
		ne = 0;
		for (i = 1; i <= n; ++i){
			for (j = 1; j <= n; ++j){
				scanf("%d", &cost);
				if (i != j){
					e[ne].u = i;
					e[ne].v = j;
					e[ne++].w = cost;
				}
			}
		}
		printf("%d\n", kruskal());
	}
	return 0;
}