POJ1861 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[N];
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;
}