HDU 3047 带权并查集

来源:互联网 发布:电脑收银软件下载 编辑:程序博客网 时间:2024/06/05 02:47

参考http://blog.csdn.net/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;}
0 0
原创粉丝点击