食物链 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

只有一个整数，表示假话的数目。

Sample Input

100 7
1 101 1 
2 1 2
2 2 3 
2 3 3 
1 1 3 
2 3 1 
1 5 5

Sample Output

3

思路：

• 一般我们都会把一个动物当成一个节点，然后去执行并查集等操作。
• 但是有位大牛另辟蹊径，给每个动物赋予三个节点（n,2n,3n）,
• 这样就将并查集的节点数量扩展到3n，用并查集维护这3n个节点的信息就行。

下面是具体的思路和操作： 

• 1 ,先来考虑什么情况下a和b两个动物是同类： 
• （1）a吃c，b也吃c，则a和b同类； 
• （2）a吃c，c吃d，d吃b，则a和b同类； 
• 2 ,有了上面的基础，我们还需要巧妙的定义下面的规则： 
• （1）如果a和b是同类，则a的三个节点(a_n,a_2n,a_3n)分别指向b的对应的三个节点(b_n,b_2n,b_3n)，
• 即a_n -> b_n , a_2n -> b_2n , a_3n -> b_3n; 
   

 

•  （2）如果a吃b，则a的三个节点(a_n,a_2n,a_3n)分别指向对应b的三个节点(b_n,b_2n,b_3n)的下一个节点，
• 即a_n -> b_2n , a_2n -> b_3n , a_3n -> b_n; 

 

•  3 ,所以我们现在可以表示1中提到的两种同类的情况了： 

(1)

 (2)

• 按照上面的规则对每一个动物去建立关系，那么，如果a和b是同类，肯定有a_n与b_n连通，
• 即有相同的根节点；如果a吃b，肯定有a_n与b_2n连通；如果b吃a，肯定有b_n与a_2n连通。 

超超爱爱---并查集模板QAQ 

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=3e5+10;
int n,k;
int f[maxn];

void init_set()
{
    for(int i=1; i<=3*n; i++)
    {
        f[i]=i;
    }
}

int find_set(int x)
{
    int r=x;
    while(r!=f[r]) r=f[r];
    int i=x;
    int j;
    while(i!=r)
    {
        j=f[i];
        f[i]=r;
        i=j;
    }
    return r;
}

void union_set(int x,int y)
{
    x=find_set(x);
    y=find_set(y);
    if(x!=y)
        f[x]=y;
}
int main()
{
    int sum=0;
    scanf("%d%d",&n,&k);
    init_set();
    for(int i=0; i<k; i++)
    {
        int x,y,a;
        scanf("%d%d%d",&a,&x,&y);
        if(x<1||y<1||x>n||y>n)
        {
            sum++;
            continue;
        }
        if(a==2&&x==y)
            sum++;
        else if(a==1)///a,b同类
        {
            if(find_set(x)==find_set(y+n)||find_set(y)==find_set(x+n)) sum++;///如果a吃b或者b吃a，说明是假话
            else///否则是真话，建立a和b同类的关系
            {
                union_set(x,y);
                union_set(x+n,y+n);
                union_set(x+2*n,y+2*n);
            }
        }
        else if(a==2)///a吃b
        {
            if(find_set(x)==find_set(y)||find_set(y)==find_set(x+n)) sum++;///如果a,b同类或者b吃a，说明是假话
            else///否则是真话，建立a吃b的关系
            {
                union_set(x,y+n);
                union_set(x+n,y+2*n);
                union_set(x+2*n,y);
            }
        }
    }
    printf("%d\n",sum);
}