您的位置:首页 > 其它

HDU 3047 带权并查集

2017-05-14 20:42 120 查看
参考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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: