并查集经典题型------食物链（java版）

并查集

操作

将两个集合合并
询问两个元素是否在一个集合中

基本原理

每个集合用一颗树表示，树根的编号就是整个集合的编号
每个节点储存它的父节点，p[x]表示x的父节点

优化

路径压缩
按秩合并

原题链接

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

解题思想

食物链具体思想：

可以用每个点到根节点的距离来表示每个点的种类关系（模3）只需要维护到根节点的距离即可

余0：表示与根节点是同类
余1：表示它可以吃根节点
余2：表示它被根节点所吃

输入如果是同类：

在一个集合中

看他们到根节点的距离，距离之差模3等于0，说明是同类，满足条件，否则就是假话

不在一个集合中

就合并两个集合，维护两个点的距离（同类）

输入不是同类，具体思想看代码

AC代码

import java.util.Scanner;

public class Main {

    static int N = 50010;
    static int[] p = new int[N];   //储存每个点的根节点
    static int[] d = new int[N];   //储存x到p[x]的距离
    static int n;
    static int m;

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();
        m = sc.nextInt();
        Init(n);//初始化操作
        int res=0;//假话个数
        while (m-- > 0) {
            int t = sc.nextInt();
            int x = sc.nextInt();
            int y = sc.nextInt();
            if(x>n||y>n){ //不满足题意，假话
                res++;
            }else{
                int px=GetRoot(x);
                int py=GetRoot(y);
                if (t==1){   //输入是同类
                    if (px==py&&(d[x]-d[y])%3!=0){//在同一个集合里,如果不是同类
                        res++;
                    }else if (px!=py){//不在同一个集合中，合并两个集合
                        p[px]=py;//将x集合合并到y集合中
                        d[px]=d[y]-d[x];  //维护距离是同类
                    }
                }else{   //不是同类
                    if (px==py&&(d[x]-d[y]-1)%3!=0){//在一个集合中，如果差值不是1
                        res++;
                    }else if(px!=py){ //不在一个集合中，合并
                        p[px]=py;
                        d[px]=d[y]+1-d[x];
                    }
                }
            }
        }
        System.out.println(res);
    }

    static void Init(int n) {
        for (int i = 1; i <= n; i++) {
            p[i] = i;
        }
    }

    static int GetRoot(int x) {
        if (x != p[x]) {
            int t = GetRoot(p[x]);//先将根节点储存下来，保证更新距离时数据不变
            d[x] += d[p[x]];   //自己到父节点的距离+父节点到根节点的距离
            p[x]=t;
        }
        return p[x];
    }
}