【Leetcode】785. Is Graph Bipartite?

题目地址：

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

判断一个$n$个点的无向图是否是二分图。

法1：DFS。从某个点开始DFS，同时顺便把点染色，如果发现矛盾则返回false。代码如下：

class Solution {
 public:
  bool isBipartite(vector<vector<int>>& g) {
    int n = g.size();
    vector<int> col(n, -1);
    for (int i = 0; i < n; i++)
      if (!~col[i] && !dfs(i, 0, col, g)) return false;
    return true;
  }

  bool dfs(int u, int c, auto& col, auto& g) {
    col[u] = c;
    for (int ne : g[u]) {
      if (~col[ne]) {
        if (col[ne] == col[u]) return false;
      } else if (!dfs(ne, c ^ 1, col, g))
        return false;
    }

    return true;
  }
};

时间复杂度$O(n+m)$，空间$O(n)$，$m$是边数。

法2：并查集。如果是二分图的话，某个点和其所有的邻接点一定要是不同的颜色，我们可以先一一判断某个点和其邻接点是否属于同一个集合，如果是，就有矛盾；同时将所有的邻接点union起来（即要标记它们为同种颜色）。遍历每个连通分量的时候，实际上最后并查集会给出一个涂色方案。代码如下：

class Solution {
 public:
  vector<int> p;

  int find(int x) { return x == p[x] ? x : p[x] = find(p[x]); }

  bool isBipartite(vector<vector<int>>& g) {
    int n = g.size();
    p.resize(n);
    for (int i = 0; i < n; i++) p[i] = i;
    for (int i = 0; i < n; i++)
      for (int ne : g[i]) {
        int pne = find(ne);
        if (find(i) == pne) return false;
        p[find(g[i][0])] = pne;
      }
    return true;
  }
};

时间复杂度$O(n{\log }^{\ast }n)$，空间$O(n)$。