****可能的二分法

给定一组 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;
    }
}

 

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值