2.判定二分图问题（染色法(DFS)）

染色法判定二分图

二分图定义：

若无向图 G=(V,E) 的顶点集 V 可以分割为两个互不相交的子集，且图中每条边的两个顶点分别属于不同的子集，则称图 G 为一个二分图。

---

785. 判断二分图

难度中等449

存在一个 无向图 ，图中有 n 个节点。其中每个节点都有一个介于 0 到 n - 1 之间的唯一编号。给你一个二维数组 graph ，其中 graph[u] 是一个节点数组，由节点 u 的邻接节点组成。形式上，对于 graph[u] 中的每个 v ，都存在一条位于节点 u 和节点 v 之间的无向边。该无向图同时具有以下属性：

• 不存在自环（graph[u] 不包含 u）。
• 不存在平行边（graph[u] 不包含重复值）。
• 如果 v 在 graph[u] 内，那么 u 也应该在 graph[v] 内（该图是无向图）
• 这个图可能不是连通图，也就是说两个节点 u 和 v 之间可能不存在一条连通彼此的路径。

二分图 定义：如果能将一个图的节点集合分割成两个独立的子集 A 和 B ，并使图中的每一条边的两个节点一个来自 A 集合，一个来自 B 集合，就将这个图称为 二分图 。

如果图是二分图，返回 true ；否则，返回 false 。

示例 1：

输入：graph = [[1,2,3],[0,2],[0,1,3],[0,2]]
输出：false
解释：不能将节点分割成两个独立的子集，以使每条边都连通一个子集中的一个节点与另一个子集中的一个节点。

示例 2：

输入：graph = [[1,3],[0,2],[1,3],[0,2]]
输出：true
解释：可以将节点分成两组: {0, 2} 和 {1, 3} 。

提示：

• graph.length == n
• 1 <= n <= 100
• 0 <= graph[u].length < n
• 0 <= graph[u][i] <= n - 1
• graph[u] 不会包含 u
• graph[u] 的所有值 互不相同
• 如果 graph[u] 包含 v，那么 graph[v] 也会包含 u

【重要】染色法(DFS)

class Solution {
    ArrayList<Integer>[] g;
    Map<Integer, Integer> map;
    public boolean isBipartite(int[][] graph) {
        int n = graph.length;
        map = new HashMap<>();
        g = new ArrayList[n];
        for(int i = 0; i < n; i++) g[i] = new ArrayList<Integer>();
        // 建图
        for(int i = 0; i < n; i++){
            for(int to : graph[i]){
                g[i].add(to);
                g[to].add(i);
            }
        }
        for(int i = 0; i < n; i++){
            // 如果当前节点还没有染色，尝试用颜色1对其染色，将相连的边都染成颜色2
            // 在对相邻边染色时，如果相邻节点颜色与节点i一致，则不符合规定，返回false ==> 最终结果就是false
            if(!map.containsKey(i) && !dfs(i, 1)) return false;
        }
        return true;
    }
    // 返回true：能成功将idx染成color，能分成两个集合
    public boolean dfs(int idx, int color){
        if(map.containsKey(idx)) return map.get(idx) == color;
        map.put(idx, color);
        for(int e : g[idx]){
            if(!dfs(e, -color)) return false;
        }
        return true;
    }
}

886. 可能的二分法

难度中等365

给定一组 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 中每一组都 不同

题解：

我们用两种颜色对图进行染色，如果可以完成染色，那么就说明可以将所有人分进两组。

具体的染色方法如下：

• 初始化所有人的颜色为 0，表示还没有染色。
• 遍历所有人，如果当前人没有染色，那么就用颜色 1 对其进行染色，然后将其所有不喜欢的人用颜色 2 进行染色。如果染色过程中出现了冲突，那么就说明无法将所有人分进两组，返回 false。
• 如果所有人都染色成功，那么就说明可以将所有人分进两组，返回 true。

class Solution {
    ArrayList<Integer>[] g;
    Map<Integer, Integer> map;
    public boolean possibleBipartition(int n, int[][] dislikes) {
        g = new ArrayList[n];
        map = new HashMap<>();
        for(int i = 0; i < n; i++){
            g[i] = new ArrayList<Integer>();
        }
        //建图
        for(int[] e : dislikes){
            g[e[0]-1].add(e[1]-1);
            g[e[1]-1].add(e[0]-1);
        }
        for(int i = 0; i < n; i++){
            if(!map.containsKey(i) && !dfs(i, 1)) return false;
        }
        return true;
    }

    public boolean dfs(int idx, int color){
        if(map.containsKey(idx)) return map.get(idx) == color;
        // 还没有染色，用颜色 1 染色
        map.put(idx, color);
        for(int next : g[idx]){
            // 将其所有不喜欢的人用颜色 2 进行染色
            if(!dfs(next, -color)) return false;
        }
        return true;
    }
}

其他

1042. 不邻接植花

难度中等198

有 n 个花园，按从 1 到 n 标记。另有数组 paths ，其中 paths[i] = [xi, yi] 描述了花园 xi 到花园 yi 的双向路径。在每个花园中，你打算种下四种花之一。

另外，所有花园 最多 有 3 条路径可以进入或离开.

你需要为每个花园选择一种花，使得通过路径相连的任何两个花园中的花的种类互不相同。

以数组形式返回 任一 可行的方案作为答案 answer，其中 answer[i] 为在第 (i+1) 个花园中种植的花的种类。花的种类用 1、2、3、4 表示。保证存在答案。

示例 1：

输入：n = 3, paths = [[1,2],[2,3],[3,1]]
输出：[1,2,3]
解释：
花园 1 和 2 花的种类不同。
花园 2 和 3 花的种类不同。
花园 3 和 1 花的种类不同。
因此，[1,2,3] 是一个满足题意的答案。其他满足题意的答案有 [1,2,4]、[1,4,2] 和 [3,2,1]

示例 2：

输入：n = 4, paths = [[1,2],[3,4]]
输出：[1,2,1,2]

示例 3：

输入：n = 4, paths = [[1,2],[2,3],[3,4],[4,1],[1,3],[2,4]]
输出：[1,2,3,4]

提示：

• 1 <= n <= 104
• 0 <= paths.length <= 2 * 104
• paths[i].length == 2
• 1 <= xi, yi <= n
• xi != yi
• 每个花园 最多 有 3 条路径可以进入或离开

哈希表实现（染色法）

题解：https://leetcode.cn/problems/flower-planting-with-no-adjacent/solution/liang-chong-xie-fa-ha-xi-biao-shu-zu-wei-7hm8/

四色定理（世界近代三大数学难题之一），又称四色猜想、四色问题，是世界三大数学猜想之一。

四色问题的内容是“任何一张地图只用四种颜色就能使具有共同边界的国家着上不同的颜色。”也就是说在不引起混淆的情况下一张地图只需四种颜色来标记就行。

用数学语言表示即“将平面任意地细分为不相重叠的区域，每一个区域总可以用1234这四个数字之一来标记而不会使相邻的两个区域得到相同的数字。”这里所指的相邻区域是指有一整段边界是公共的。如果两个区域只相遇于一点或有限多点就不叫相邻的。

• 问题相当于用 4 种颜色给图中的每个节点染色，要求相邻节点颜色不同。而「所有花园最多有 3 条路径可以进入或离开」，这相当于图中每个点的度数至多为 3，那么只要选一个和邻居不同的颜色即可。

class Solution {
    public int[] gardenNoAdj(int n, int[][] paths) {
        List<Integer> g[] = new ArrayList[n];
        Arrays.setAll(g, e -> new ArrayList<>());
        for(int[] e : paths){
            int x = e[0]-1, y = e[1]-1; // 编号改为从0开始
            g[x].add(y);
            g[y].add(x);
        }
        int[] color = new int[n];
        for(int i = 0; i < n; i++){
            // 至多有4种颜色，每个节点至多3个路径
            // 因此只要选与邻居不同的颜色即可
            boolean[] used = new boolean[5];
            for(int j : g[i]){
                used[color[j]] = true;
            }
            while(used[++color[i]]); // 找到一个邻居没有用过的颜色
            // for color[i]++; used[color[i]]; color[i]++ {}
        }
        return color;
    }
}