畅游图论之判断二分图

最新推荐文章于 2024-03-28 10:28:28 发布

阿飞算法

最新推荐文章于 2024-03-28 10:28:28 发布

阅读量172

点赞数

分类专栏： Graph Algorithm LeetCode

本文链接：https://blog.csdn.net/wat1r/article/details/119518065

版权

LeetCode 同时被 2 个专栏收录

113 篇文章 12 订阅

订阅专栏

Graph Algorithm

27 篇文章 4 订阅

订阅专栏

畅游图论之判断二分图

方法1:BFS

public boolean isBipartite(int[][] graph) {
    int V = graph.length;
    int[] colors = new int[V];
    Arrays.fill(colors, -1);//-1表示未着色
    Queue<Integer> queue = new LinkedList<>();
    for (int i = 0; i < V; i++) {
        if (colors[i] == -1) {//当前点未着色
            queue.offer(i);
            colors[i] = 0;//着色0
            while (!queue.isEmpty()) {
                int u = queue.poll();
                for (int v : graph[u]) {
                    if (colors[v] == colors[u]) return false;
                    if (colors[v] == -1) {
                        colors[v] = 1 - colors[u];//着色u的颜色的相反色 只有两种颜色， 0和1 
                        queue.offer(v);
                    }
                }
            }
        }
    }
    return true;
}

另外一种写法

public boolean isBipartite(int[][] graph) {
    int V = graph.length;
    int[] colors = new int[V];//默认0为未着色
    Queue<Integer> queue = new LinkedList<>();
    for (int i = 0; i < V; i++) {
        if (colors[i] == 0) {//当前点未着色
            queue.offer(i);
            colors[i] = 1;//着色1
            while (!queue.isEmpty()) {
                int u = queue.poll();
                for (int v : graph[u]) {
                    if (colors[v] == colors[u]) return false;
                    if (colors[v] == 0) {
                        colors[v] = -colors[u];//着色u的颜色的相反色 只有两种颜色， -1和1
                        queue.offer(v);
                    }
                }
            }
        }
    }
    return true;
}

方法2:DFS

public boolean isBipartite(int[][] graph) {
    int V = graph.length;
    int[] colors = new int[V];//默认0为未着色
    for (int i = 0; i < V; i++) {
        if (colors[i] == 0 && !helper(graph, colors, i, 1)) return false;
    }
    return true;
}


/**
 * @param graph  图
 * @param colors colors数组
 * @param u      当前要处理的顶点
 * @param c      要为u着色的颜色color
 * @return
 */
private boolean helper(int[][] graph, int[] colors, int u, int c) {
    if (colors[u] != 0) {//当前顶点u没有被着色，要给其安排颜色c，如果符合安排，说明当前的着色方案没问题，反之则不可以
        return colors[u] == c;
    }
    colors[u] = c;//着色
    for (int v : graph[u]) {
        if (!helper(graph, colors, v, -c)) return false;//着色失败，则提前退出
    }
    return true;
}