Forsaken喜欢独一无二的树

Forsaken喜欢独一无二的树

https://ac.nowcoder.com/acm/contest/1221/H

题意:给定一个图,让最小生成树唯一,问最小去掉边权和。
思路:在kruskal的思想下,重复的边一定是和已经合并两个集合之间的边权相等的。然后我看了大佬代码发现可以双指针?

#include<bits/stdc++.h>
using namespace std;
const int N=400010;
typedef long long ll;
int p[N];
int find(int x)
{
    if(x!=p[x])
        p[x]=find(p[x]);
    return p[x];
}
struct node{
    int a;
    int b;
    int c;
    bool operator<(const node&W)const
    {
        return c<W.c;
    }
}q[N];
int main()
{
    int n,m;
    cin>>n>>m;
    for(int i=0;i<m;i++)
    {
        cin>>q[i].a>>q[i].b>>q[i].c;
    }
    sort(q,q+m);
    for(int i=1;i<=n;i++)
        p[i]=i;
    ll res=0;
    for(int i=0;i<m;i++)
    {
        int j=i+1;
        while(j<m && q[j].c==q[i].c)     j++;
        for(int k=i;k<j;k++)
        {
            int a=q[k].a;
            int b=q[k].b;
            if(find(a)!=find(b))
                res+=q[k].c;
        }
        for(int k=i;k<j;k++)
        {
            int a=q[k].a;
            int b=q[k].b;
            if(find(a)!=find(b))
            {
                p[find(a)]=find(b);
                res-=q[k].c;
            }
        }
        i=j-1;
    }
    cout<<res<<endl;
    return 0;
}
全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务