POJ2377 用并查集实现最大生成树（速度较快）

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

#define SIZN 1001
#define SIZM 20001

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

int p[SIZN];
edge e[SIZM];//边集数组
int rank[SIZN];
int total=0,t=0;
bool cmp(edge e1, edge e2)   //快排
{
    return e1.w > e2.w;
}

int findset(int x)   //找最终父节点
{
    if (x!=p[x])   //p[x]是x的父节点，（根节点的父节点就是自己）
    {
        p[x]=findset(p[x]);
    }
    return p[x];
}

int Union(int yh, int jh)
{
    int rx, ry;
    int x,y;
    x=findset(yh);
    y=findset(jh);
    if (rank[y]>rank[x])
    {
        p[x]=y;
    }
    else{
        p[y]=x;
        if (rank[y]==rank[x])
            rank[y]++;
    }
    return 0;
}
int main()
{
    int i,n,m;
    scanf("%d%d",&n,&m);
    total=0;
    memset(p,0,sizeof(p));
    memset(rank,0,sizeof(rank));
    for (i=1;i<=n;i++)   //初始化：每个节点的父节点就是自己本身
        p[i]=i;
    for (i=1;i<=m;i++)
    {
        scanf("%d%d%d" , &e[i].u , &e[i].v , &e[i].w );
    }
    sort(&e[1],&e[m+1],cmp);   //排序 ，前面两个参数分别为待排序数组是首地址和尾地址

    for (i=1;i<=m ;i++)
    {
        if (findset(e[i].u)!=findset(e[i].v))
        {
            Union(e[i].v,e[i].u);
            t++;
            total+=e[i].w;
        }
    }
    if (t<n-1)printf("%d/n",-1);    //生成树的边数最小为n-1，不然不可能使所有节点都连通
    else printf("%d/n",total);
    system("pause");
    return 0;
}