最小生成树-kruskal算法

#include <iostream>
#include <algorithm>
using namespace std;
const int N=150;
int father[N],r[N],u[N],v[N],w[N];//定义father为每个节点的父亲，r为每条节点的编号，u,v为节点，w为边的权重
int find(int x)//并查集的核心算法，找寻父亲节点
{
    int r=x;
    while(father[r]!=r)
    {
        r=father[r];
    }
    return r;
}
bool cmp(const int a,const int b)//主要用于sort函数中对于边的权重的排序
{
    return w[a]<w[b];
}
int kruskal(int n,int m)//核心算法
{
    int mst=0,cnt=0;
    for(int i=0;i<n;i++)//初始化每个节点的父亲为自己本身
    {
        father[i]=i;
    }
    for(int i=0;i<m;i++)//记录每条边的序号
    {
        r[i]=i;
    }
    sort(r,r+m,cmp);//将边按照权重从小到大排序，这里只改变序号，就可以知道是哪条边
    for(int i=0;i<m;i++)//对每条边来说，都判断边的两个顶点是否连通，如果连通，则证明加入之后，会出现回路，不予考虑
    {
        int e=r[i];
        int x=find(u[e]);
        int y=find(v[e]);
        if(x!=y)
        {
            mst+=w[e];
            father[x]=y;
            cnt++;
        }
    }
    if(cnt<n-1)//证明此图不连通，不存在最小生成树
        mst=0;
    return mst;
}
int main()
{
    int n,m;
    cin>>n>>m;
    for(int i=0;i<m;i++)
    {
        cin>>u[i]>>v[i]>>w[i];
    }
    int mst=kruskal(n,m);
    if(mst==0)
        cout<<"the mst is not found"<<endl;
    else
        cout<<"the mst is  "<<mst<<endl;
    return 0;
}