0并查集中等 LeetCode886. 可能的二分法

886. 可能的二分法

描述

给定一组 n 人（编号为 1, 2, …, n）， 我们想把每个人分进任意大小的两组。每个人都可能不喜欢其他人，那么他们不应该属于同一组。

给定整数 n 和数组 dislikes ，其中 dislikes[i] = [ai, bi] ，表示不允许将编号为 ai 和 bi的人归入同一组。当可以用这种方法将所有人分进两组时，返回 true；否则返回 false。

示例 1：

输入：n = 4, dislikes = [[1,2],[1,3],[2,4]]
输出：true
解释：group1 [1,4], group2 [2,3]
示例 2：

输入：n = 3, dislikes = [[1,2],[1,3],[2,3]]
输出：false
示例 3：

输入：n = 5, dislikes = [[1,2],[2,3],[3,4],[4,5],[1,5]]
输出：false
 

提示：

1 <= n <= 2000
0 <= dislikes.length <= 104
dislikes[i].length == 2
1 <= dislikes[i][j] <= n
ai < bi
dislikes 中每一组都 不同

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/possible-bipartition
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

分析

并查集
一个数字所有他对应的数字都在一个小组里面，即在并查集里是相互连接的，否则就冲突了。
计算邻接数组，然后从1遍历到n，判断一个数字时候与他对应的数字存在连接，如果存在说明在一个小组，即冲突了返回false。

class Solution {
    int[] parents;
    public boolean possibleBipartition(int n, int[][] dislikes) {
        parents = new int[n+1];
        List<Integer>[] adjoin = new List[n+1];
        for (int i = 0; i < n+1; i++) {
            adjoin[i] = new ArrayList<>();
        }
        for (int i = 1; i < n+1; i++) {
            parents[i] = i;
        }
        for (int[] arr : dislikes) {
            adjoin[arr[0]].add(arr[1]);
            adjoin[arr[1]].add(arr[0]);
        }

        for (int i = 1; i <= n; i++) {
            for (int j = 0; j < adjoin[i].size(); j++) {
                if (find(i) == find(adjoin[i].get(j))) {
                    return false;
                }
                union(adjoin[i].get(j),adjoin[i].get(0));
            }
        }
        return true;
    }

    public int find(int i) {
        if (i != parents[i]) {
            parents[i] = find(parents[i]);
        }
        return parents[i];
    }

    public void union(int i, int j) {
        int a = find(i);
        int b = find(j);
        parents[a] = b;
    }
}