HDU - 3047 Zjnu Stadium(带权并查集)

题目链接https://vjudge.net/contest/346897#problem/B
在这里插入图片描述
翻译
n个观众标号1~n,坐在环形的观众席上,一个位置可以坐好几个人。
给出 a b x 即b在a的顺时针x距离处。
现在给出你m条信息,让你判断错误的有几条(和前面冲突的就算错)
并查集维护节点到祖先的距离。
代码

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=50010;
int f[N],sum[N];
int getf(int x)
{
    if(x==f[x])
        return x;
    int temp=f[x];/*此处一定要把f[x]的值赋给一个变量,要不然下面f[x]的值就会发生改变*/
    f[x]=getf(f[x]);
    sum[x]=sum[x]+sum[temp];
    return f[x];
}
int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        for(int i=1; i<=n; i++)
        {
            f[i]=i;
            sum[i]=0;
        }
        int a,b,x,num=0;
        while(m--)
        {
            scanf("%d%d%d",&a,&b,&x);
            int t1=getf(a);
            int t2=getf(b);
            if(t1==t2)
            {
                if(sum[a]+x!=sum[b])/*祖先相等,直接比较距离*/
                    num++;
            }
            else
            {
                f[t2]=t1;
                sum[t2]=sum[a]+x-sum[b];
            }
        }
        printf("%d\n",num);
    }
    return 0;
}
发布了165 篇原创文章 · 获赞 6 · 访问量 5056

猜你喜欢

转载自blog.csdn.net/lylzsx20172018/article/details/103432175
今日推荐