poj1182 并查集经典题 Weighted Union-Find Sets

没注意只有一个case 不知道怎么就错了 - -

带权并查集的"权"表示该节点与爸爸的关系

这题的关系有

0.儿子与爸爸是同类

1.儿子被爸爸是吃 - -

2.儿子吃爸爸 = =

为什么权要这样设?待会解释

所以每个节点多一个变量记录权

那两个不是父子关系的点之间的关系怎么表示呢?

为了解决这个问题 我们把每条指向父亲的边都看作向量!

eg:假如现在UFS是这样的

(节点,权) (1,0)->(2,1)->(3,2)->(4,0)

那么可知:

1与2的关系是同类;

2被3吃;

3吃4;

4与4(自己)是同类

2与1是什么关系呢? 因为1->2的长度(权)是0 那么2->1的长度不也是0吗? 就是说2与1也是同类

3与2是什么关系? 同理 |3->2|=-1 怎么办? 权只有 0,1,2 啊 ! 所以-1+3=2 即权为2 就是3吃吃2了

如果 a吃b b吃c c吃d 呢? 是多少怎么处理?

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 50010;

struct node
{
int f,r;
}ani[maxn];

inline int mod(int a)
{
a%=3;
return a<0?a+3:a;
}

int ff(int x)
{
int &f=ani[x].f,t=f;
f=(f==x?x:ff(f));
ani[x].r=mod(ani[x].r+ani[t].r);    ///BE CAREFUL!  Father is change!!
return f;
}

int main()
{
//freopen("a.in","r",stdin);
int m,n;
scanf("%d%d",&m,&n);
int ans=0;
for(int i=1;i<=m;i++)
{
ani[i].f=i;
ani[i].r=0;
}
for(int i=0;i<n;i++)
{
int o,a,b;
scanf("%d%d%d",&o,&a,&b);
o--;
int ra=ff(a),rb=ff(b);  ///a to be father of b
if(a>m||b>m)
ans++;
else if(o&&a==b)
ans++;
else
{
if(ra!=rb)
{
ani[rb].f=ra;
ani[rb].r=mod(o+ani[a].r-ani[b].r);
}
else if(mod(ani[b].r-ani[a].r)!=o)
ans++;
}
}
printf("%d\n",ans);
return 0;
}