题目解读:对题目的把握需要抓住一共只有三种动物这一点,这很重要,我们可以假设x吃y,y就是x的父亲结点,那么一直延续下去,我们就可以发现x%3 == y%3同类,x%3 = (y+1)%3 x吃y这一点了,有了维护路径这一点,就需要使用带权并查集了
带权并查集我们这个题的写法是只有Find函数,并且在Find函数中更新路径,嫁接两颗树的时候,我们只做对原本是根的点进行距离更新,再通过下一次访问这些点时用到的Find更新所有点。
在这里重点说如何更新原本为根的点的距离
情况一:x与y是同类生物,dx%3==dy%3,需要满足这个等式,xy才是同类,我们假设x到新根的距离为nx,原来的距离为dx,y到根的距离为dy,为满足等式,我们直接可以列出nx = dy这一等式,又有nx = dpx+dx,dpx为旧根到新根的距离
情况二:同理情况一
动物王国中有三类动物A,B,C,这三类动物的食物链构成了有趣的环形。A吃B, B吃C,C吃A。
现有N个动物,以1-N编号。每个动物都是A,B,C中的一种,但是我们并不知道它到底是哪一种。
有人用两种说法对这N个动物所构成的食物链关系进行描述:
第一种说法是"1 X Y",表示X和Y是同类。
第二种说法是"2 X Y",表示X吃Y。
此人对N个动物,用上述两种说法,一句接一句地说出K句话,这K句话有的是真的,有的是假的。当一句话满足下列三条之一时,这句话就是假话,否则就是真话。
1) 当前的话与前面的某些真的话冲突,就是假话;
2) 当前的话中X或Y比N大,就是假话;
3) 当前的话表示X吃X,就是假话。
你的任务是根据给定的N(1 <= N <= 50,000)和K句话(0 <= K <= 100,000),输出假话的总数。
Input
第一行是两个整数N和K,以一个空格分隔。
以下K行每行是三个正整数 D,X,Y,两数之间用一个空格隔开,其中D表示说法的种类。
若D=1,则表示X和Y是同类。
若D=2,则表示X吃Y。
Output
只有一个整数,表示假话的数目。
Sample Input
100 7 1 101 1 2 1 2 2 2 3 2 3 3 1 1 3 2 3 1 1 5 5
Sample Output
3
#include<cstdio>
#include<iostream>
using namespace std;
#define MAXN 50015
int f[MAXN];
int d[MAXN];
int k;
int n;
int op,u,v;
int fu,fv;
int ans = 0;
int Find(int x)
{
if(f[x] != x)
{
int p = Find(f[x]);
d[x] += d[f[x]];
f[x] = p;
}
return f[x];
}
int main()
{
cin>>n>>k;
for(int i = 1; i <= n; i++) f[i] = i;
while(k--)
{
scanf("%d %d %d",&op,&u,&v);
if(u>n||v>n)
{
ans++;
//cout<<u<<" "<<v<<endl;
}
else
{
fu = Find(u),
fv = Find(v);
if(op == 1)
{
if(fu == fv && (d[u]-d[v])%3)
{
ans++;
// cout<<u<<" "<<v<<endl;
}
else if(fu != fv)
{
f[fu] = fv;
d[fu] = d[v] - d[u];
}
}
else
{
if(fu == fv && (d[u]-d[v]-1)%3)
{
ans++;
//cout<<d[u]<<" "<<d[v]<<endl;
//cout<<u<<" "<<v<<endl;
}
else if(fu != fv)
{
f[fu] = fv;
d[fu] = d[v] - d[u] + 1;
}
}
}
//printf("f[1] = %d f[2] = %d f[3] = %d\n",f[1],f[2],f[3]);
//printf("d[1] = %d d[2] = %d d[3] = %d\n",d[1],d[2],d[3]);
}
printf("%d\n",ans);
return 0;
}