把所以节点分为三类
1.到根节点的距离为0(第0代)
2.到根节点的距离为1(第1代)
3.到根节点的距离为2(第2代)
代数就是到根节点的距离%3,第3代和第0代属于同一类,第1代可以吃根节点,第2代被根节点所吃
题目描述
动物王国中有三类动物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和K句话,输出假话的总数。
输入格式
第一行是两个整数N和K,以一个空格分隔。
以下K行每行是三个正整数 D,X,Y,两数之间用一个空格隔开,其中D表示说法的种类。
若D=1,则表示X和Y是同类。
若D=2,则表示X吃Y。
输出格式
只有一个整数,表示假话的数目。
数据范围
1≤N≤50000,
0≤K≤100000
样例
输入样例:
100 7
1 101 1
2 1 2
2 2 3
2 3 3
1 1 3
2 3 1
1 5 5
输出样例:
3
AC 代码
#include <iostream>
using namespace std;
const int N = 50001;
int n,m,k;
int op,x,y;
int fa[N],d[N];
int find (int x)
{
if (fa[x] != x)
{
int t = find(fa[x]); // fa[x]不能先变
d[x] += d[fa[x]]; // x到fa[x]的距离加上fa[x]到祖宗的距离(因为此时fa[x]没变)
fa[x] = t;
}
return fa[x];
}
int main()
{
cin >> n >> m;
int ans = 0;
for (int i = 1; i <= n; i++)
fa[i] = i;
while ( m-- )
{
cin >> op >> x >> y;
if (x > n || n < y) ans ++;
else
{
int xx = find (x),yy = find(y);
if (op == 1)
{
if (xx == yy && (d[x] - d[y]) % 3) ans ++; // 如果x和y在同一颗数上,并且代数不同
else if (xx != yy)
{
fa[xx] = yy;
d[xx] = d[y] - d[x]; // 因为(d[x] + d[xx] - d[y]) % 3 = 0,so d[xx] = d[y] - d[x];
}
}
else
{
if (xx == yy && (d[x] - d[y] - 1) % 3) ans ++; // 如果x不是比y多一代
else if (xx != yy)
{
fa[xx] = yy;
d[xx] = d[y] + 1 - d[x]; // 因为(d[x] + d[xx] - 1 - d[y]) % 3 = 0,so d[xx] = d[y] + 1 - d[x];
}
}
}
}
cout << ans;
return 0;
}