num.14食物链 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(1 <= N <= 50,000)和K句话(0 <= K <= 100,000),输出假话的总数。
Input
第一行是两个整数N和K,以一个空格分隔。
以下K行每行是三个正整数 D,X,Y,两数之间用一个空格隔开,其中D表示说法的种类。
若D=1,则表示X和Y是同类。
若D=2,则表示X吃Y。
Output
只有一个整数,表示假话的数目。

样例

num.14食物链 POJ - 1182_第1张图片

大体思想:

1、如果两个物种有联系,不管是吃,被吃还是同类,它们之间应该是有一条径路可达的,
也就是它们在一个合集中。
2、如果a,b有关系,b,c有关系,那么a,c之间的关系式可以通过两者的关系推出来的。
分为以下三点
1、新加的关系表明x,y是同类,那么它们两个分别到连通区域根点节的关系应该是一样的,要不就矛盾了。
2、如果新加的关系表明x,y不是同类,那么在当前参加y,x相对根节点的关系和x本来相对根点节的关系应该是不变的,否则就矛盾了。
3、两者在不同的连通集里头:就直接连接两个连通集就能够了。

代码

#include 
#include 
using namespace std;
int pre[50005];
int arr[50005];//根点节到点节的关系

void init(int n)
{
     
    for(int i=0;i<=n;i++)
    {
     
        pre[i]=i;
        arr[i] = 0;
    }
}
int find(int x)
{
     
    if(x!=pre[x])//不是根点节
    {
     
        int temp = pre[x];//将当前点节的父点节设置为根点节
        pre[x] = find(temp);
        arr[x] = (arr[x]+arr[temp]) % 3;
    }
    return pre[x];
}

int main()
{
     
    int n,m,x,y,d,fx,fy,num;
    while(~scanf("%d %d",&n,&m))//POJ上只要需一次入输,所以不要需while循环
    {
     
        num= 0;
        init(n);
        for(int i=0;i<m;i++)
        {
     
            scanf("%d %d %d",&d,&x,&y);
            if(x>n||y>n)
            {
     
                num++;
                continue;
            }
            if(d==2&&x==y)
            {
     
                num++;
                continue;
            }
            fx = find(x);
            fy = find(y);
            if(fx == fy)//属于同一个子集
            {
     
                //如果x、y是同类,那么他们相对根点节的关系应该是一样的
                if(d==1 && arr[x]!=arr[y])
                    num++;
                if(d==2 && arr[x]!=(arr[y]+2)%3)
                    num++;
            }
            else//合并两个连通区域
            {
     
                pre[fy] = fx;//y根的父点节更新成x根
                arr[fy] = (arr[x]+(d-1)+(3-arr[y])) % 3;//注意这里只更新的是fy相对于根的关系
            }
        }
        printf("%d\n", num);
    }
    return 0;
}

PS
https://www.cnblogs.com/dongsheng/archive/2013/06/12/3133188.html

你可能感兴趣的:(并查集)