POJ - 1258 Agri-Net (最小生成树)

题目链接：点击打开链接

题意：求一棵树的最小生成树。

题解：prime 和 Kurskal二选一跑一遍=-=。 多组输入。赤裸裸的一道模板题，就一点也不用改=-=。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define size 1000
using namespace std;
const int maxn = 1e6+5;
struct node{
	int u;
	int v;
	int w;
}edge[maxn];
int n,cnt;
int f[maxn];
bool cmp(node a,node b){
	return a.w < b.w;
}
int find(int x){
	if(x == f[x]) return x;
	else return f[x] = find(f[x]);
}
void kruskal(){
	sort(edge,edge+cnt,cmp);
	for(int i = 0 ; i < n + 5; i ++)
		f[i] = i;
	int sum = 0;
	int num = 0;
	for(int i = 0 ; i < cnt ; i ++){
		int u = find(edge[i].u);
		int v = find(edge[i].v);
		if(u != v){
			f[u] = v;
			sum += edge[i].w;
			num ++;
			if(num == n - 1){
				printf("%d\n",sum);
				return ;
			}
		}
	}
}
int main(){

	while(~scanf("%d",&n)){
		cnt = 0;
		for(int i = 0 ; i < n ; i ++){
			for(int j = 0 ; j < n ; j ++){

				scanf("%d",&edge[cnt].w);
				if(i == j) continue;
				edge[cnt].u = i;
				edge[cnt].v = j;
				cnt ++;
			}
		}
		kruskal();
	}
	return 0;
}