NC17509 挖沟

\(MST\)板子题

const int N=5e5+10;
struct Node
{
    int a,b,c;
    bool operator<(const Node &W) const
    {
        return c<W.c;
    }
}e[N];
int p[N];
int n,m;

int find(int x)
{
    if(x != p[x]) p[x]=find(p[x]);
    return p[x];
}

int kruskal()
{
    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=e[i].a,b=e[i].b,c=e[i].c;
        int pa=find(a),pb=find(b);
        if(pa != pb)
        {
            p[pa]=pb;
            res+=c;
            cnt++;
            if(cnt == n-1) return res;
        }
    }
}


int main()
{
    ios;
    cin>>n>>m;

    for(int i=0;i<m;i++)
    {
        int a,b,c;
        cin>>a>>b>>c;
        e[i]={a,b,c};
    }

    sort(e,e+m);

    int t=kruskal();
    cout<<t<<endl;

    //system("pause");
}
原文地址:https://www.cnblogs.com/fxh0707/p/13795824.html