基于并查集的kruskal算法

#include <iostream>            //并查集的kruskal算法
using namespace std;
const int max_ve=1005,max_ed=15005;
int n,m,i;    //n,m分别记录顶点数和边数
struct node
{
    int par,ans;
}vertex[max_ve];    //顶点
struct Edge 
{
    int u,v,weigh;
}edge[max_ed];        //边
int cmp(const void* a,const void* b)
{
    return (*(Edge*)a).weigh-(*(Edge*)b).weigh;
}
int find_p(int j)        //在处理两个顶点是否在同个集合内要用到并查集
{
    if(vertex[j].par!=j)
    {
        vertex[j].par=find_p(vertex[j].par);
    }
    return vertex[j].par;
}
bool union_set(int s,int t)
{
    int os=find_p(s),ot=find_p(t);
    if(os==ot)
        return false;
    if(vertex[os].ans>vertex[ot].ans)
        vertex[ot].par=os;
    else
        vertex[os].par=ot;
    if(vertex[os].ans==vertex[ot].ans)
        vertex[ot].ans++;
    return true;
}
int main()
{
    int sum,ct;
    scanf("%d%d",&n,&m);        
    for(i=1;i<=n;++i)    //初始化，顶点从1到n
    {
        vertex[i].par=i;vertex[i].ans=0;
    }
    for(i=0;i<m;++i)    //输入边的信息：起点，终点，权重
    {
        scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].weigh);        
    }
    qsort(edge,m,sizeof(edge[0]),cmp);
    sum=0;ct=0;
    for(i=0;i<m&&ct<n-1;++i)    //当ct==n-1说明有n-1条边，即所有的顶点都已连接上，终止算法
    {
        if(union_set(edge[i].u,edge[i].v))
        {
            ++ct;    //边数+1
            sum+=edge[i].weigh;
        }
    }
    printf("%d\n",sum);    //sum为最小生成树所有边的总和
    return 0;
}