洛谷-P2024-食物链

题目描述

动物王国中有三类动物 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 句话有的是真的，有的是假的。当一句话满足下列三条之一时，这句话就是假话，否则就是真话。

• 当前的话与前面的某些真的话冲突，就是假话；
• 当前的话中 X 或 Y 比 N 大，就是假话；
• 当前的话表示 X 吃 X，就是假话。

你的任务是根据给定的 N 和 K 句话，输出假话的总数。

输入格式

第一行两个整数，N,K，表示有 N 个动物，K 句话。

第二行开始每行一句话（按照题目要求，见样例）

输出格式

一行，一个整数，表示假话的总数。

输入输出样例

输入 

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

输出 

3

说明/提示

对于全部数据，1≤N≤5×10^4，1≤K≤10^5。

思路分析：

        动物之间有三种关系：同类，捕食，被捕食。题目要我们求给出的所有关系中的假关系（假话），所有假话可以分为三种：所说的动物超出编号；假如上面的话中有x吃y了，后面再出现y吃x或y和x同类等相矛盾的话；动物自己吃自己。

        本题显然是一个带权并查集问题，对于上述动物的三种关系我们可以用数字012分别表示。剩下的就是经典的两步操作：查找(find)和合并(unsion)，所谓查找就是路径压缩，将就是在搜索的时候找到最远的祖先，然后将父亲节点赋值，对于权值而言，就是找出权值与最远祖先之间所有边权传递的过程，找出节点与父亲节点的关系，依次传递即可。对于合并操作，就需要通过画图找到两节点各自与根节点的关系，假设两点间权值为x，列出权值方程求出x。

        显然带权并查集是复杂的，但是本题可以有更简单的方法解决。

        我们可以采用种类并查集，开3*N大小的集合，其中1-N表示同类关系，N+1-2*N表示捕食关系，2*N+1-3*N表示被捕食关系。遍历每一句话，如果当前这句话符合上述假话中的其中一种，便假话计数加1，如果当前不是假话，则将其归并到话中所说对应的集合中去。

代码如下：

import java.util.Scanner;

public class 食物链 {
    static int N = 50010;
    static int[] fa = new int[3 * N];
    static int cnt = 0;
    // 并查集查找模板
    static int find(int x){
        if(x == fa[x]) return x;
        return fa[x] = find(fa[x]);
    }
    // 并查集合并模板
    static void unison(int x, int y){
        x = find(x); y = find(y);
        if(x != y) fa[x] = y;
    }
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt(), k = sc.nextInt();
        for(int i = 1; i <= 3 * n; i++) fa[i] = i;
        while(k-- > 0){
            int op = sc.nextInt(), x = sc.nextInt(), y = sc.nextInt();
            // 超出编号一定是假话
            if(x > n || y > n){
                cnt++;
                continue;
            }
            if(op == 1){ // xy同类：判断x的猎物中有没有y、x的天敌中有没有y
                if(find(x + n) == find(y) || find(x + 2 * n) == find(y)){
                    cnt++;
                    continue;
                }
                // 是真话则归并所有xy为同类的集合
                unison(x, y); unison(x + n, y + n); unison(x + 2 * n, y + 2 * n);
            }
            if(op == 2){ // x吃y：判断x和y是不是同类、x的天敌中有没有y
                if(find(x) == find(y) || find(x + 2 * n) == find(y)){
                    cnt++;
                    continue;
                }
                // 真话归并所有x吃y的集合
                unison(x + n, y); unison(x + 2 * n, y + n); unison(y + 2 * n, x);
            }
        }
        System.out.println(cnt);
    }
}