力扣 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 先将dislikes分组 利用List将不能与其同组的人记录在g[i]中

2 看当前元素是否染色 若没有染色，则dfs对其染色

3 dfs中判断能否染色成功 若不喜欢的人已经染色 且与不喜欢的人染同样的颜色 则染色失败

若不喜欢的人没有染色 则对不喜欢的人染另外一种颜色 看其能否染色成功

class Solution {
    public boolean possibleBipartition(int n, int[][] dislikes) {
        // 染色标志数组
      int[] color = new int[n+1];
        // 记录i不喜欢的人
        List<Integer>[] g = new List[n+1];
        // 将g的每个项初始化为空数组
        for(int i = 0;i <= n; i++) {
            g[i] = new ArrayList<Integer>();
        }
        // 对g数组初始化 g[i]中存放i不喜欢的人
        for(int[] p : dislikes) {
            g[p[0]].add(p[1]);
            g[p[1]].add(p[0]);
        }
        // 对没有染色的进行染色 染色失败则返回false
        for(int i =1;i <= n;i++){
            if(color[i] == 0 && !dfs(i,1,color,g)) {
                return false;
            }
        }
        return true;
    }
    // dfs判断能否染色成功
    public boolean dfs(int i,int nowcolor,int[] color,List<Integer>[] g) {
       color[i] = nowcolor;
    //    循环不喜欢的人
    for(int p:g[i]) {
        // 已染色 且染同样的颜色 染色失败
        if(color[p] != 0 && color[p] == color[i]) {
            return false;
        }
        // 若不喜欢的人没有染色 则对其染不同的颜色 看能否染色成功
        if(color[p] == 0 && !dfs(p,3^nowcolor,color,g)) {
            return false;
        }
    }
    return true;
    }
}