并查集 - 食物链 - POJ - 1182

并查集 - 食物链 - 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

题解：
$$\begin{gathered} 共有3种生物，捕食关系形成一个环，那么， \\ 对任意三只动物，知道任意两只动物之间的关系，就能够推断出三只动物之间相互的关系。 \\ 如：若x吃y，y吃z，则z吃x。或者，若x吃y，y与z是同类，则x吃z。 \end{gathered}$$

$$\begin{gathered} 这样，对任意两个节点，只需要查询它们与根节点的的关系，即可推断出这两个节点之间的关系。 \\ 因此，我们用一个数组来记录各节点到根节点的距离，将距离对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]。 \end{gathered}$$

另外：

$$\begin{gathered} 在建立并查集同时压缩路径的过程中，需要加上dis数组的更新过程。 \\ x到根的距离即x到父节点p[x]的距离+p[x]到根节点的距离，需要临时保存根节点tmp，先更新距离，再压缩路径。 \end{gathered}$$

---

代码：

#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;
}