用染色法判断二分图

用染色法来染色节点， 如果途中没有出现矛盾的点，那么说明该无向图是二分图。
伪代码如下

for(int i = 1; i <= n; i++)
	if(i未被遍历过)
		dfs(i);

题目

代码

#include<cstring>
#include<algorithm>
#include<iostream>

using namespace std;

const int N = 100010, M = 200010;
int n, m;
int e[M], ne[M], idx, h[N];
int color[N];

void add(int a, int b){
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
}

bool dfs(int u, int c){//节点u， 颜色为c
    color[u] = c;
    
    for(int i = h[u]; i != -1; i = ne[i]){
        int j = e[i];
        if(!color[j]){
            if(!dfs(j, 3 - c)) return false;
        }
        else if(color[j] == c) return false;
    }
    return true;
}
int main(){
    cin >> n >>m;
    
    memset(h, -1, sizeof h);
    
    while(m -- ){
        int a, b;
        cin >> a >>b;
        add(a, b), add(b, a);
    }
    
    //二分染色法
    bool flag = true;
    for(int i = 1; i <= n; i++){
        if(!color[i])
            if(!dfs(i, 2)){ 
                flag = false;
                break;
            }
    }
    if(flag) puts("Yes");
    else puts("No");
    return 0;
}