并查集优化最小生成树

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=10010;
int n,m;
int p[N];//存储并查集
struct Edge
{
    int a,b,w;
    bool operator<(const Edge &W)const
    {
        return w<W.w;
    }
}edges[N];
int find(int x)//并查集
{
    if(x!=p[x])p[x]=find(p[x]);
    return p[x];
}

int main()
{
    cin>>n>>m;
    for(int i=0;i<m;i++)
    {
        int a,b,w;
        cin>>a>>b>>w;
        edges[i]={a,b,w};
    }
    sort(edges,edges+m);

    for(int i=1;i<n;i++)p[i]=i;

    int res=0;//树的长度
    int cnt=0;//边的个数

    for(int i=0;i<m;i++)
    {
        int a=edges[i].a,b=edges[i].b,w=edges[i].w;

        a=find(a),b=find(b);
        if(a!=b)
        {
            p[a]=b;
            res+=w;
            cnt++;
        }
    }
    if(cnt<n-1)puts("impossible");
    else cout<<res;
    return 0;
}