POJ#1182 食物链

食物链

Time Limit: 1000MS		Memory Limit: 10000K
Total Submissions: 100235		Accepted: 30292

Description

动物王国中有三类动物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

Source

Noi 01

 

题意：如上。

思路：建立一个三段的数组1~3*N；其中1~N表示i∈A，N+1~2*N表示i∈B，2*N+1~3*N表示i∈C。

　　首先初始化3*N大小的并查集，再依据每次给出的语句进行判断或更行并查集。

　　定义find函数时Par[a]==a;漏了个=，bug了一上午。。💩

　　scanf和cin在本题就是AC和TLE的区别

#include<cstdio>
#include<algorithm>
using namespace std;
int Par[3 * 50000+10],Rank[3*50000+10];
void init(int n) {
    for (int i = 1; i <= n; i++) {        //Node从1开始
        Par[i] = i;
    }
    fill(Rank + 1, Rank + n + 1, 0);
}
int find(int a) {
    if (Par[a] == a)return a;
    else return Par[a] = find(Par[a]);
}
bool same(int a, int b) {
    return find(a) == find(b);
}
void unit(int x, int y) {
    x = find(x);
    y = find(y);
    if (x == y)return;
    else if (Rank[x] < Rank[y])Par[x] = y;
    else if (Rank[x] > Rank[y])Par[y] = x;
    else {
        Par[y] = x;
        Rank[x]++;
    }
}
int main() {
    int N, K,t,x,y,ans=0;
    scanf("%d%d",&N,&K);
    init(3 * N);
    while (K--) {
        scanf("%d%d%d",&t,&x,&y);
        if (x > N|| y>N) {
            ans++;
            continue;
        }
        if (t == 1) {
            if (same(x, y + N) || same(x, y + 2 * N) ){
                ans++;
            }
            else {
                unit(x, y);
                unit(x + N, y + N);
                unit(x + 2 * N, y + 2 * N);
            }
        }
        else {
            if (same(x, y) || same(x, y+2*N)) {
                ans++;
            }
            else {
                unit(x, y + N);
                unit(x + N, y + N * 2);
                unit(x + 2 * N, y);
            }
        }
    }
    printf("%d\n", ans);
    return 0;
}