leetcode-判断二分图

. - 力扣（LeetCode）

存在一个 无向图 ，图中有 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 。

示例 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} 。

提示：

• 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

分析：对graph遍历，逐渐将所有节点分到2个集合中。用不同的flag来标记集合A、B即可。例如划分到集合A标记为1，划分到B的标记为-1，没有划分的标记为0

例如[[1,2,3],[0,2],[0,1,3],[0,2]]：

遍历第0个，将0分到集合A中，那[1,2,3]只能分到集合B中；

遍历第1个，1已经在集合B中，那[0,2]只能分到集合A中，但是2已经分到了集合B中，所以划分失败

class Solution {
public:
    bool isBipartite(vector<vector<int>>& graph) {
        unordered_set<int> a;
        unordered_set<int> b;
        for (int i = 0; i < graph.size(); i++) {
            auto& nodes = graph[i];
            if (a.find(i) != a.end()) {
                for (int j = 0; j < nodes.size(); j++) {
                    if (a.find(nodes[j]) != a.end()) {
                        return false;
                    }
                    if (b.find(nodes[j]) == b.end()) {
                        b.insert(nodes[j]);
                    }
                }                
            } else if (b.find(i) != b.end()) {
                for (int j = 0; j < nodes.size(); j++) {
                    if (b.find(nodes[j]) != b.end()) {
                        return false;
                    }
                    if (a.find(nodes[j]) == a.end()) {
                        a.insert(nodes[j]);
                    }
                }
            } else {
                a.insert(i);
                for (int j = 0; j < nodes.size(); j++) {
                    if (a.find(nodes[j]) != a.end()) {
                        return false;
                    }
                    if (b.find(nodes[j]) == b.end()) {
                        b.insert(nodes[j]);
                    }
                }
            }
        }
        return true;
    }
};

class Solution {
public:
    bool isBipartite(vector<vector<int>>& graph) {
        int n = graph.size();
        vector<int> flags(n, 0);
        for (int i = 0; i < n; i++) {
            auto& nodes = graph[i];
            if (flags[i] == -1) {// in A
                for (int j = 0; j < nodes.size(); j++) {
                    if (flags[nodes[j]] == -1) {// in A
                        return false;
                    }
                    flags[nodes[j]] = 1;// put in B
                }
            } else if (flags[i] == 1) {// in B
                for (int j = 0; j < nodes.size(); j++) {
                    if (flags[nodes[j]] == 1) {// in B
                        return false;
                    }
                    flags[nodes[j]] = -1;// put in A
                }                
            } else {
                flags[i] = -1;
                for (int j = 0; j < nodes.size(); j++) {
                    if (flags[nodes[j]] == -1) {// in A
                        return false;
                    }
                    flags[nodes[j]] = 1;// put in B
                }                
            }
        }
        return true;
    }
};

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

    bool isBipartite(int curNode, int curFlag, vector<int>& flags, vector<vector<int>>& graph) {
        if (flags[curNode] != 0) {
            return flags[curNode] == curFlag;
        }
        flags[curNode] = curFlag;
        for (auto nextNode : graph[curNode]) {
            if (!isBipartite(nextNode, -curFlag, flags, graph)) {
                return false;
            }
        }
        return true;
    }
};