leetcode886. 可能的二分法

题目

给定一组 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,2][1,3][2,4]可以表示为：

遍历每一行，得到相邻节点；例如想要找到1的相邻节点，遍历第一行，得到2、3；如果将1染成红色的话，需要将2、3染成蓝色，染色过程如下：

Code

class Solution {
    public boolean possibleBipartition(int n, int[][] dislikes) {
        // 邻接矩阵
        int[][] matrix = new int[n + 1][n + 1];
        for(int[] item : dislikes){
            matrix[item[0]][item[1]] = 1;
            matrix[item[1]][item[0]] = 1;
        }
        // 相邻节点染色
        int[] record = new int[n + 1];
        for(int i = 1; i <= n; i++){
            // 只看未染色的节点
            if(record[i] == 0 && !dfs(matrix, record, i, 1, n)) return false;
        }
        return true;
    }

    private boolean dfs(int[][] matrix, int[] record, int idx, int color, int n){
        record[idx] = color;
        for(int i = 1; i <= n; i++){
            if(i == idx) continue;
            // 相邻节点, 颜色与当前节点相同
            if(matrix[idx][i] == 1 && record[i] == color) return false; 
            // 相邻节点, 未染色, 先尝试染色(dfs)
            if(matrix[idx][i] == 1 && record[i] == 0 && !dfs(matrix, record, i, -1 * color, n)) return false;
        }

        return true;
    }
}

结果