给定一组 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
DFS或者扩展并查集
并查集:
class Solution {
public boolean possibleBipartition(int N, int[][] dislikes) {
UnionFind uf = new UnionFind(N * 2 + 1);
for(int[] in : dislikes) {
int a = uf.find(in[0]);
int b = uf.find(in[1]);
int c = uf.find(in[0] + N);
int d = uf.find(in[1] + N);
if(uf.find(a) == uf.find(b)) return false;
uf.merge(b,c);
uf.merge(a,d);
}
return true;
}
class UnionFind{
int[] parent;
public UnionFind(int x) {
parent = new int[x];
for(int i = 0; i < x; i++) parent[i] = i;
}
public int find(int x) {
while(parent[x] != x) {
x = parent[x];
}
return x;
}
public void merge(int x, int y) {
int a = find(x);
int b = find(y);
if(a == b) return;
else parent[a] = b;
}
}
}
DFS:
class Solution {
List<List<Integer>> mat = new ArrayList<List<Integer>>();
Map<Integer, Integer> color = new HashMap<Integer, Integer>();
public boolean possibleBipartition(int N, int[][] dislikes) {
for(int i = 0; i <= N; i++) mat.add(new ArrayList<Integer>());
for(int i = 0; i < dislikes.length; i++) {
mat.get(dislikes[i][0]).add(dislikes[i][1]);
mat.get(dislikes[i][1]).add(dislikes[i][0]);
}
for(int i = 1; i <= N; i++) {
if(!color.containsKey(i) && !dfs(i, 0)) return false;
}
return true;
}
private boolean dfs(int i, int c) {
if(color.containsKey(i)) return color.get(i) == c;
color.put(i, c);
int cn = c == 0 ? 1 : 0;
for(int t : mat.get(i)) {
if(!dfs(t, cn)) return false;
}
return true;
}
}