leetcode 785.判断二分图 Java

做题博客链接

https://blog.csdn.net/qq_43349112/article/details/108542248

题目链接

https://leetcode-cn.com/problems/is-graph-bipartite/

描述

存在一个 无向图 ，图中有 n 个节点。其中每个节点都有一个介于 0 到 n - 1 之间的唯一编号。给你一个二维数组 
graph ，其中 graph[u] 是一个节点数组，由节点 u 的邻接节点组成。形式上，对于 graph[u] 中的每个 v ，都存
在一条位于节点 u 和节点 v 之间的无向边。该无向图同时具有以下属性：
不存在自环（graph[u] 不包含 u）。
不存在平行边（graph[u] 不包含重复值）。
如果 v 在 graph[u] 内，那么 u 也应该在 graph[v] 内（该图是无向图）
这个图可能不是连通图，也就是说两个节点 u 和 v 之间可能不存在一条连通彼此的路径。
二分图 定义：如果能将一个图的节点集合分割成两个独立的子集 A 和 B ，并使图中的每一条边的两个节点一个来自 
A 集合，一个来自 B 集合，就将这个图称为 二分图 。

如果图是二分图，返回 true ；否则，返回 false 。



提示：

graph.length == n
1 <= n <= 100
0 <= graph[u].length < n
0 <= graph[u][i] <= n - 1
graph[u] 不会包含 u
graph[u] 的所有值 互不相同
如果 graph[u] 包含 v，那么 graph[v] 也会包含 u

示例

示例 1：

输入：graph = [[1,2,3],[0,2],[0,1,3],[0,2]]
输出：false
解释：不能将节点分割成两个独立的子集，以使每条边都连通一个子集中的一个节点与另一个子集中的一个节点。

示例 2：

输入：graph = [[1,3],[0,2],[1,3],[0,2]]
输出：true
解释：可以将节点分成两组: {0, 2} 和 {1, 3} 。

初始代码模板

class Solution {
    public boolean isBipartite(int[][] graph) {

    }
}

代码

推荐题解：
https://leetcode-cn.com/problems/is-graph-bipartite/solution/bfs-dfs-bing-cha-ji-san-chong-fang-fa-pan-duan-er-/

BFS

class Solution {
    public boolean isBipartite(int[][] graph) {
        Queue<Integer> queue = new LinkedList<>();
        //0表示没访问过，1和-1标识子集A和子集B
        int[] vis = new int[graph.length];

        for (int i = 0; i < graph.length; i++) {
            if (vis[i] != 0) {
                //说明已经访问过
                continue;
            }

            vis[i] = 1;
            queue.offer(i);
            //System.out.println(Arrays.toString(vis));
            while (!queue.isEmpty()) {
                int num = queue.poll();
                for (int v : graph[num]) {
                    if (vis[v] != 0 && vis[v] != -vis[num]) {
                        return false;
                    } else if (vis[v] == 0) {
                        vis[v] = -vis[num];
                        queue.offer(v);
                    }
                }
            }
           
        }

        return true;
    }
}

DFS

class Solution {
    public boolean isBipartite(int[][] graph) {
        int n = graph.length;
        int[] vis = new int[n];

        for (int i = 0; i < n; i++) {
            if (vis[i] != 0) {
                continue;
            }

            vis[i] = 1;
            if (!dfs(graph, i, vis)) {
                return false;
            }
        }

        return true;
    }

    private boolean dfs(int[][] graph, int u, int[] vis) {
        for (int v : graph[u]) {
            if (vis[v] == vis[u]) {
                return false;
            }
            if (vis[v] == 0) {
                vis[v] = -vis[u];
                if (!dfs(graph, v, vis)) {
                    return false;
                }
            }
        }

        return true;
    }
}