并查集的普通应用包括,计算最小生成树的kruscal算法和无向图的连通性问题。
这个题目是并查集的一个扩展应用,用于解决等价类问题。
问题描述:
动物王国中有三类动物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),输出假话的总数。
输入:
第一行是两个整数N和K,以一个空格分隔。
以下K行每行是三个正整数 D,X,Y,两数之间用一个空格隔开,其中D表示说法的种类。
若D=1,则表示X和Y是同类。
若D=2,则表示X吃Y。
输出:
只有一个整数,表示假话的数目。
算法思路:
在原有parent[]的基础上加上另一关系relative[x],用于表示动物x相对于其根节点的关系:
1. 如果relative[x]=0, 说明x和根节点的关系是同类
2. 如果relative[x]=1, 说明x和根节点的关系是x吃根节点
3. 如果relative[x]=2, 说明x被根节点吃。
由于再输入的时候也会输入x和y的关系D,题目中说D=1表示x和y是同类,D=2表示x吃y,与relative[]关系表示有出入,所以在输入D之后要做一写调整D-1。
下面对比较难理解的地方进行讲解:
1.
首先通过find()函数找出x和y的根节点rootx和rooty,
如果rootx和rooty相等,那么说明x和y的关系在原来语句的基础上是可以判断对错的,if(d==1&&animal[x].relative!=animal[y].relative) 如果输入的关系是d=1,说明这句话表明x和y是同类,而通过判断animal[x].relative与animal[y].relative是否是同类,如果animal[x].relative!=animal[y].relative说明在原来语句的基础上推出x和y不是同类,这样就与当前这个语句矛盾了,所以错误数++。
if(d==2&&(animal[x].relative-animal[y].relative+3)%3!=1),如果输入关系d=2,说明x是吃y的,也就是说x相对于y的关系是1,那么怎么从以前语句中推出x和y的关系呢?是(animal[x].relative-animal[y].relative+3)%3,具体解释得通过向量关系来推导:
x相对于rootx的关系向量是animal[x].relative,y相对于rooty的关系向量是animal[y].relative,要求出x相对于y的关系,=》(animal[x].relative-animal[y].relative+3)%3,如果这个值和1不相等,也就是说以前的语句和当前语句矛盾,所以错误数++.
如果rootx和rooty不相等,说明x和y的关系在原有语句的基础上是不能确定的,于是我们就认为这个语句是正确的,于是调用Union(rootx,rooty,x,y,d-1);把x节点的树和y节点的树合并。注意这里为什么是d-1,是因为在前面解释了。
2.
在union()函数中,除了要对树rootx和树rooty进行合并,同时还要对rooty节点的relative做修改。
这里直接就是把rooty的parent指向了rootx.
虽然parent关系已经转了,但是relative关系还没有修改,注意这里只是修改了rooty相对于rootx的关系,而对于rooty的所有子节点相对于rootx的关系是没有修改的,这个修改时在find()函数中做的。那么怎么对rooty的relative关系进行修改呢?
animal[RootOfY].relative = (animal[NodeX].relative-animal[NodeY].relative-D+3)%3;
向量解释如下:
X相对于y的关系已经给了是D,y相对于rooty的关系是animal[y].relative, x相对于rootx的关系是animal[x].relative,那么rooty相对于rootx的关系就是(animal[NodeX].relative-animal[NodeY].relative-D+3)%3;
3.
在find函数中除了要返回nodeTofind的根节点,同时还要对在union()中没有修改的节点的relative关系做修改.
首先用temp存储nodetoFind的父节点,之后递归调用Find()函数,对nodefoFind的父节点的relative做修改,也就是说find()完之后,nodetoFind父节点相对于根节点的关系已经确定了animal[tmp].relative,现在还知道nodetoFind相对于temp节点的关系animal[NodetoFind].relative。
于是可以计算nodetofind相对于root的关系:
Animal[nodetofind].relative = (animal[temp].relative + animnal[nodetofind].relative)%3