并查集 - 食物链 - POJ - 1182
题目:
动物王国中有三类动物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
题解:
共
有
3
种
生
物
,
捕
食
关
系
形
成
一
个
环
,
那
么
,
对
任
意
三
只
动
物
,
知
道
任
意
两
只
动
物
之
间
的
关
系
,
就
能
够
推
断
出
三
只
动
物
之
间
相
互
的
关
系
。
如
:
若
x
吃
y
,
y
吃
z
,
则
z
吃
x
。
或
者
,
若
x
吃
y
,
y
与
z
是
同
类
,
则
x
吃
z
。
共有3种生物,捕食关系形成一个环,那么,\\对任意三只动物,知道任意两只动物之间的关系,就能够推断出三只动物之间相互的关系。\\如:若x吃y,y吃z,则z吃x。或者,若x吃y,y与z是同类,则x吃z。
共有3种生物,捕食关系形成一个环,那么,对任意三只动物,知道任意两只动物之间的关系,就能够推断出三只动物之间相互的关系。如:若x吃y,y吃z,则z吃x。或者,若x吃y,y与z是同类,则x吃z。
这 样 , 对 任 意 两 个 节 点 , 只 需 要 查 询 它 们 与 根 节 点 的 的 关 系 , 即 可 推 断 出 这 两 个 节 点 之 间 的 关 系 。 因 此 , 我 们 用 一 个 数 组 来 记 录 各 节 点 到 根 节 点 的 距 离 , 将 距 离 对 3 取 模 , 得 到 的 余 数 0 、 1 、 2 分 别 对 应 三 种 关 系 : d i s [ x ] % 3 = 0 : x 与 根 是 同 类 , d i s [ x ] % 3 = 1 : x 吃 根 , d i s [ x ] % 3 = 2 : x 被 根 吃 。 ① 、 当 D = 1 时 , 二 者 是 同 类 , 若 x 与 y 的 祖 先 相 同 , 说 明 之 前 已 经 建 立 关 系 , 此 时 需 要 判 断 是 否 冲 突 , 判 断 d i s [ x ] % 3 = d i s [ y ] % 3 是 否 成 立 , 即 判 断 ( d i s [ x ] − d i s [ y ] ) % 3 = 0 是 否 成 立 。 若 x 与 y 的 祖 先 不 同 , 此 时 需 要 建 立 关 系 , 可 以 把 x 的 祖 先 接 在 y 的 祖 先 之 下 , 同 时 d i s 数 组 更 新 : 设 此 时 x 到 根 节 点 的 距 离 为 d i s [ x ] + t , t 是 x 原 先 的 祖 先 p x 接 到 y 的 祖 先 p y 下 时 , 需 要 加 上 的 距 离 , 那 么 有 ( d i s [ x ] + t − d i s [ y ] ) % 3 = 0 , 于 是 可 以 取 t = d i s [ y ] − d i s [ x ] 。 ② 、 当 D = 2 时 , x 吃 y , 若 x 与 y 的 祖 先 相 同 , 说 明 之 前 已 经 建 立 关 系 , 此 时 需 要 判 断 是 否 冲 突 , 判 断 d i s [ x ] % 3 = d i s [ y ] % 3 + 1 是 否 成 立 。 即 判 断 ( d i s [ x ] − 1 − d i s [ y ] ) % 3 = 0 是 否 成 立 。 若 x 与 y 的 祖 先 不 同 , 此 时 需 要 建 立 关 系 , 同 样 把 x 的 祖 先 接 在 y 的 祖 先 之 下 , 同 时 d i s 数 组 更 新 : 设 此 时 x 到 根 节 点 的 距 离 为 d i s [ x ] + t , 那 么 有 ( d i s [ x ] + t − 1 − d [ y ] ) % 3 = 0 , 于 是 取 t = d [ y ] + 1 − d [ x ] 。 这样,对任意两个节点,只需要查询它们与根节点的的关系,即可推断出这两个节点之间的关系。\\\ \\因此,我们用一个数组来记录各节点到根节点的距离,将距离对3取模,得到的余数0、1、2分别对应三种关系:\\dis[x]\%3=0:x与根是同类,dis[x]\%3=1:x吃根,dis[x]\%3=2:x被根吃。\\ \ \\①、当D=1时,二者是同类,\\\qquad若x与y的祖先相同,说明之前已经建立关系,此时需要判断是否冲突,判断dis[x]\%3=dis[y]\%3是否成立,\\即判断(dis[x]-dis[y])\%3=0是否成立。\\\qquad若x与y的祖先不同,此时需要建立关系,可以把x的祖先接在y的祖先之下,同时dis数组更新:\\设此时x到根节点的距离为dis[x]+t,t是x原先的祖先px接到y的祖先py下时,需要加上的距离,\\那么有(dis[x]+t-dis[y])\%3=0,于是可以取t=dis[y]-dis[x]。\\\ \\②、当D=2时,x吃y,\\\qquad若x与y的祖先相同,说明之前已经建立关系,此时需要判断是否冲突,判断dis[x]\%3=dis[y]\%3+1是否成立。\\即判断(dis[x]-1-dis[y])\%3=0是否成立。\\\qquad若x与y的祖先不同,此时需要建立关系,同样把x的祖先接在y的祖先之下,同时dis数组更新:\\设此时x到根节点的距离为dis[x]+t,那么有(dis[x]+t-1-d[y])\%3=0,于是取t=d[y]+1-d[x]。 这样,对任意两个节点,只需要查询它们与根节点的的关系,即可推断出这两个节点之间的关系。 因此,我们用一个数组来记录各节点到根节点的距离,将距离对3取模,得到的余数0、1、2分别对应三种关系:dis[x]%3=0:x与根是同类,dis[x]%3=1:x吃根,dis[x]%3=2:x被根吃。 ①、当D=1时,二者是同类,若x与y的祖先相同,说明之前已经建立关系,此时需要判断是否冲突,判断dis[x]%3=dis[y]%3是否成立,即判断(dis[x]−dis[y])%3=0是否成立。若x与y的祖先不同,此时需要建立关系,可以把x的祖先接在y的祖先之下,同时dis数组更新:设此时x到根节点的距离为dis[x]+t,t是x原先的祖先px接到y的祖先py下时,需要加上的距离,那么有(dis[x]+t−dis[y])%3=0,于是可以取t=dis[y]−dis[x]。 ②、当D=2时,x吃y,若x与y的祖先相同,说明之前已经建立关系,此时需要判断是否冲突,判断dis[x]%3=dis[y]%3+1是否成立。即判断(dis[x]−1−dis[y])%3=0是否成立。若x与y的祖先不同,此时需要建立关系,同样把x的祖先接在y的祖先之下,同时dis数组更新:设此时x到根节点的距离为dis[x]+t,那么有(dis[x]+t−1−d[y])%3=0,于是取t=d[y]+1−d[x]。
另外:
在 建 立 并 查 集 同 时 压 缩 路 径 的 过 程 中 , 需 要 加 上 d i s 数 组 的 更 新 过 程 。 x 到 根 的 距 离 即 x 到 父 节 点 p [ x ] 的 距 离 + p [ x ] 到 根 节 点 的 距 离 , 需 要 临 时 保 存 根 节 点 t m p , 先 更 新 距 离 , 再 压 缩 路 径 。 在建立并查集同时压缩路径的过程中,需要加上dis数组的更新过程。\\x到根的距离即x到父节点p[x]的距离+p[x]到根节点的距离,需要临时保存根节点tmp,先更新距离,再压缩路径。 在建立并查集同时压缩路径的过程中,需要加上dis数组的更新过程。x到根的距离即x到父节点p[x]的距离+p[x]到根节点的距离,需要临时保存根节点tmp,先更新距离,再压缩路径。
代码:
#include<iostream>
#include<string>
#include<cstdio>
using namespace std;
const int N=5e4+10;
int n,k,d,x,y,p[N],cnt,dis[N];
int find(int x)
{
if(p[x]!=x)
{
int tmp=find(p[x]);
dis[x]+=dis[p[x]];
p[x]=tmp;
}
return p[x];
}
int main()
{
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++) p[i]=i;
while(k--)
{
scanf("%d%d%d",&d,&x,&y);
if(x>n||y>n) cnt++;
else
{
int px=find(x),py=find(y);
if(d==1)
{
if(px==py&&(dis[x]-dis[y])%3) cnt++;
else if(px!=py)
{
p[px]=py;
dis[px]=dis[y]-dis[x];
}
}
else
{
if(px==py&&(dis[x]-dis[y]-1)%3) cnt++;
else if(px!=py)
{
p[px]=py;
dis[px]=dis[y]+1-dis[x];
}
}
}
}
printf("%d\n",cnt);
return 0;
}