HDU 3047 带权并查集

最新推荐文章于 2020-01-04 22:03:19 发布

原创最新推荐文章于 2020-01-04 22:03:19 发布 · 271 阅读

0 ·

CC 4.0 BY-SA版权

并查集专栏收录该内容

14 篇文章

订阅专栏

本文介绍了一种使用带权并查集解决位置规则冲突问题的方法。通过具体实例讲解了如何判断规则是否正确，并利用向量思想进行树的合并，有效避免了位置重复的问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

参考http://blog.youkuaiyun.com/shuangde800/article/details/7983965
题意：有n个人（1…n），然后给你m条规则，每条规则格式为：a,b,x。表示第a个人的位置+x是第b个人的位置。然后问你有几个是不正确的规则（看作一维）
分析：有几个规则是不正确的，也就是说不能有两个人的位置相同（一定）
我们可以用带权值的并查集来做，如果a，b根结点不同，则规则是正确的，将a，b两个树合并，如果a，b的根节点是相同的，则看他们满不满足第a个人的位置+x是第b个人的位置，若不满足，则是错误的规则，否为正确的
那么问题来了，如果ab的根结点不同，那怎么将之合并呢，用向量的思想

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
const int maxn=100000 + 10;
//带权并查集
int f[maxn];
int r[maxn];
int ans=0;
int findf(int k)//找爸爸，以及压缩路径及位置
{
   if(f[k]==k) return k;
   else
   {
       int temp=f[k];
       f[k]=findf(f[k]);
       r[k]+=r[temp];
   }
   return f[k];
}

int main()
{
    int n,m;
    while(scanf("%d %d",&n,&m)!=EOF)
    {
        ans=0;
        for(int i=0;i<=n;i++) {f[i]=i;r[i]=0;}
        for(int i=0;i<m;i++)
        {
            int a,b,x;
            scanf("%d %d %d",&a,&b,&x);
            int root1=findf(a);
            int root2=findf(b);
            if(root1!=root2)
            {
                f[root2]=root1;
                r[root2]=r[a]+x-r[b];//合并
            }
            else {
                if((r[a]+x)!=r[b])
                    ans++;
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}