# 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 <= n <= 2000
  • 0 <= dislikes.length <= $10^4$
  • dislikes[i].length == 2
  • 1 <= dislikes[i][j] <= n
  • ai < bi
  • dislikes 中每一组都 不同

# 题解

染色法加深搜

class Solution {
public:
bool dfs(int curnode, int nowcolor, vector<int>& color, const vector<vector<int>>& g) {
color[curnode] = nowcolor;
for (auto& nextnode : g[curnode]) {
if (color[nextnode] && color[nextnode] == color[curnode]) {
return false;
}
if (!color[nextnode] && !dfs(nextnode, 3 ^ nowcolor, color, g)) {
return false;
}
}
return true;
}

bool possibleBipartition(int n, vector<vector<int>>& dislikes) {
vector<int> color(n + 1, 0);
vector<vector<int>> g(n + 1);
for (auto& p : dislikes) {
g[p[0]].push_back(p[1]);
g[p[1]].push_back(p[0]);
}
for (int i = 1; i <= n; ++i) {
if (color[i] == 0 && !dfs(i, 1, color, g)) {
return false;
}
}
return true;
}
};

复杂度分析

  • 时间复杂度:$O (n+m)$
  • 空间复杂度:$O (n+m)$