题目:食物链
描述:
动物王国中有三类动物 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 句话有的是真的,有的是假的。
当一句话满足下列三条之一时,这句话就是假话,否则就是真话。
- 当前的话与前面的某些真的话冲突,就是假话;
- 当前的话中 X 或 Y 比 N 大,就是假话;
- 当前的话表示 X 吃 X,就是假话。
你的任务是根据给定的 NN 和 KK 句话,输出假话的总数。
输入格式
第一行是两个整数 N 和 K,以一个空格分隔。
以下 KK 行每行是三个正整数 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
原题链接:https://www.acwing.com/problem/content/242/
代码段及注释:
#include<iostream>
using namespace std;
const int N = 50010;
int p[N], d[N];//存父节点、存到根节点的距离
int find(int x)//找祖源、压缩路径
{
if(p[x] != x)
{
int u = find(p[x]); //为了更新深度,先把祖借点存下来((?))
d[x] += d[p[x]];//更新到根节点的距离(即深度)
p[x] = u;
}
return p[x];
}
int main()
{
int n, m;
scanf("%d%d", &n, &m);
for(int i = 1; i <= n; i ++) p[i] = i;
int res = 0;
while(m --)
{
int t, x, y;
scanf("%d%d%d", &t, &x, &y);
if(x > n || y > n) res ++;//超数量上限直接为假话
else
{
int px = find(x), py = find(y);
if(t == 1)
{
if(px == py && (d[x] - d[y]) % 3) res ++;//在同一树中通过到根的距离判断关系
else if(px != py)//不在同一树中则创造为同类关系
{
p[px] = py;
d[px] = d[y] - d[x];//求连通后两个祖节点之间的距离
}
}
else//原理同上,只是两者之间的距离改变
{
if(px == py && (d[x] - d[y] - 1) % 3) res ++;
else if(px != py)
{
p[px] = py;
d[px] = d[y] + 1 - d[x];
}
}
}
}
printf("%d\n", res);
return 0;
}