染色法判定是否是二分图
作者:互联网
算法思路:使用深度优先遍历遍历整张图,如果发现某个顶点已经染过色,并且和相邻顶点的颜色相同,则返回false,否则一直染下去,直到遍历完整张图。
代码实现:
#include <iostream> #include <cstring> #include <algorithm> using namespace std; const int N = 100010, M = 2 * N; int h[N], e[M], ne[M], idx; int color[N]; int n, m; bool flag = true; void add(int a, int b) { e[idx] = b; ne[idx] = h[a]; h[a] = idx++; } void dfs(int u, int c) { color[u] = c; for(int i=h[u];i!=-1;i=ne[i]) { int j = e[i]; if(!color[j]) { dfs(j, 3-c); } else { if(color[j] == c) { flag = false; break; } } } } int main() { memset(h, -1,sizeof h); cin >> n >> m; while(m--) { int u, v; cin >> u >> v; add(u, v); add(v, u); } for(int i=1;i<=n;i++) { // 使用for循环的原因是图可能不联通 if(!color[i]) dfs(i, 2); // 只有没染过色才能染,染1或者染2都行 } if(flag) puts("Yes"); else puts("No"); return 0; }
代码来自ACWing(https://www.acwing.com/)。
标签:二分,idx,int,是否是,ne,color,add,染色法,include 来源: https://www.cnblogs.com/cvxer/p/16473419.html