leetcode 886.可能的二分法 Java

做题博客链接

https://blog.csdn.net/qq_43349112/article/details/108542248

题目链接

https://leetcode-cn.com/problems/possible-bipartition/

描述

给定一组 N 人（编号为 1, 2, ..., N）， 我们想把每个人分进任意大小的两组。

每个人都可能不喜欢其他人，那么他们不应该属于同一组。

形式上，如果 dislikes[i] = [a, b]，表示不允许将编号为 a 和 b 的人归入同一组。

当可以用这种方法将所有人分进两组时，返回 true；否则返回 false。


提示：

1 <= N <= 2000
0 <= dislikes.length <= 10000
dislikes[i].length == 2
1 <= dislikes[i][j] <= N
dislikes[i][0] < dislikes[i][1]
对于 dislikes[i] == dislikes[j] 不存在 i != j

示例

示例 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

初始代码模板

class Solution {
    public boolean possibleBipartition(int N, int[][] dislikes) {

    }
}

代码

其实这个图，可以转换为无向图的环路判定，和二分图的检测十分相似

推荐官方的DFS题解：
https://leetcode-cn.com/problems/possible-bipartition/solution/ke-neng-de-er-fen-fa-by-leetcode/

我这里使用的BFS，其实差不多

class Solution {
    public boolean possibleBipartition(int N, int[][] dislikes) {
        //0表示未被访问，1和-1表示不同的分组
        int[] vis = new int[N + 1];
        List<Integer>[] list = new List[N + 1];
        for (int i = 1; i <= N; i++) {
            list[i] = new ArrayList<>();
        }

        for (int[] d : dislikes) {
            list[d[0]].add(d[1]);
            list[d[1]].add(d[0]);
        }

        Queue<Integer> queue = new LinkedList<>();
        for (int i = 1; i <= N; i++) {
            if (vis[i] != 0) {
                continue;
            }

            vis[i] = 1;
            queue.offer(i);
            while (!queue.isEmpty()) {
                int u = queue.poll();

                for (int v : list[u]) {
                    if (vis[v] == vis[u]) {
                        return false;
                    }
                    if (vis[v] == 0) {
                        vis[v] = -vis[u];
                        queue.offer(v);
                    }
                }
            }
        }

        return true;
    }
}