LeetCode 886.可能的二分法

题目

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

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

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

当可以用这种方法将每个人分进两组时，返回 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

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

题解

将所有人分为A，B两组，那么每个人不是A组的就一定是B组的。只有当这样分配时是不存在冲突的，那么我们才可以认定是符合题目要求的。所以我们将所有的人进行染色，用 -1 和 1 来标记当前分组。对所有人开始遍历并染色,如果发现不可行则返回false.

class Solution {
public:
    vector<int> colors;
    map<int,vector<int>> mp;
    bool dfs(int i , int color){
        //设置当前人物的颜色
        colors[i] = color;
        for(int index : mp[i]){
            if(colors[index] == color) // 和下个人物颜色相同
                return false;
            if(colors[index] == 0 && !dfs(index,-color))    //未被染色则进行染色
                return false;
        }
        return true;
    }
    bool possibleBipartition(int N, vector<vector<int>>& dislikes) {
        if(dislikes.size() == 0 )
            return true;
        colors.resize(N+1);
        // 记录不共存的值
        for(vector<int> dislike : dislikes){
            mp[dislike[0]].push_back(dislike[1]);
            mp[dislike[1]].push_back(dislike[0]);
        }
        //开始染色
        for(int i = 1; i <= N; i ++){
            if(colors[i] == 0 && !dfs(i,1))
                return false;
        }
        return true;
    }
};