POJ1251 Jungle Roads(Kruskal)(并查集）

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <climits>
#include <cstring>
#include <string>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <vector>
#include <list>
#include<functional>
#define mod 1000000007
#define inf 0x3f3f3f3f
#define pi acos(-1.0)
using namespace std;
typedef long long ll;
const int N=1005;
const int M=15005;

struct Edg {
int v,u;
int w;
} edg[M];
bool cmp(Edg g,Edg h) {
return g.w<h.w;
}
int n,m,maxn,cnt;
int parent
;
void init() {
for(int i=0; i<n; i++)parent[i]=i;
}
void Build() {
int w,k;
char u,v;
for(int i=0; i<n-1; i++) {
cin>>u>>k;
while(k--) {
cin>>v>>w;
edg[++cnt].u=u-'A';
edg[cnt].v=v-'A';
edg[cnt].w=w;
}
}
sort(edg,edg+cnt+1,cmp);
}
int Find(int x) {
if(parent[x] != x) parent[x] = Find(parent[x]);
return parent[x];
}//查找并返回节点x所属集合的根节点
void Union(int x,int y) {
x = Find(x);
y = Find(y);
if(x == y) return;
parent[y] = x;
}//将两个不同集合的元素进行合并
void Kruskal() {
int sum=0;
int num=0;
int u,v;
for(int i=0; i<=cnt; i++) {
u=edg[i].u;
v=edg[i].v;
if(Find(u)!=Find(v)) {
sum+=edg[i].w;
num++;
Union(u,v);
}
//printf("!!!%d %d\n",num,sum);
if(num>=n-1) {
printf("%d\n",sum);

break;
}
}
}
int main() {
while(~scanf("%d",&n)&&n) {
cnt=-1;
init();
Build();
Kruskal();
}
return 0;
}