本文涉及知识点
C++图论
C++DFS
LeetCode785. 判断二分图
存在一个 无向图 ,图中有 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
DFS
m_ans[i]为0表示未处理,1在A组,2在B组。
DFS(cur),next是cur的临接点,
iGroup = (3-m_ans[cur])
if( iGroup+ m_ans[next] == 3) 则不是二分图。
for i = 0 To n-1
if(m_ans[i]>0)continue;
m_ans[i]=1
DFS(i)
由于存在环,所以无法通过父节点除重,需要vis数组避免重复处理,否则就死循环了。不如:用BFS。
代码
核心代码
class Solution {
public:
bool isBipartite(vector<vector<int>>& graph) {
m_vAns.resize(graph.size());
m_vis.resize(graph.size());
for (int i = 0; i < graph.size(); i++) {
if (m_vAns[i] > 0) { continue; }
m_vAns[i] = 1;
DFS(i, graph);
}
return m_ans;
}
void DFS(int cur, vector<vector<int>>& graph) {
if (m_vis[cur]) { return; }
m_vis[cur] = true;
for (const auto& next : graph[cur]) {
const int iGroup = 3 - m_vAns[cur];
if (iGroup + m_vAns[next] == 3) { m_ans = false; }
m_vAns[next] = iGroup;
DFS(next, graph);
}
}
bool m_ans = true;
vector<int> m_vAns;
vector<bool> m_vis;
};
核心代码
vector<vector<int>> graph;
TEST_METHOD(TestMethod1)
{
graph = { {0},{2,3},{1,3},{1,2} };
auto res = Solution().isBipartite(graph);
AssertEx(false, res);
}
TEST_METHOD(TestMethod11)
{
graph = { {1,2,3},{0,2},{0,1,3},{0,2} };
auto res = Solution().isBipartite(graph);
AssertEx(false, res);
}
TEST_METHOD(TestMethod12)
{
graph = { {1,3},{0,2},{1,3},{0,2} };
auto res = Solution().isBipartite(graph);
AssertEx(true, res);
}
TEST_METHOD(TestMethod13)
{
graph = { {},{2,4,6},{1,4,8,9},{7,8},{1,2,8,9},{6,9},{1,5,7,8,9},{3,6,9},{2,3,4,6,9},{2,4,5,6,7,8} };
auto res = Solution().isBipartite(graph);
AssertEx(false, res);
}