AcWing 860. 染色法判定二分图

代码：

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

using namespace std;

const int N = 100010, M = 200010;//无向图，边数为顶点数2倍

int n, m;
int h[N], e[M], ne[M], idx;
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) {
    color[u] = c;
    
    //遍历所有邻边
    for (int i = h[u]; i != -1; i = ne[i]) {
        int j = e[i];
        if (!color[j]) {
            //当前为1号色，则3-c染为2号色，当前2号色时同理
            if (!dfs(j, 3 - c)) {
                return false;
            } 
        } else if (c == color[j]) {
                return false;//相邻点着同一种颜色
        }
    }
    
    return true;
}

int main() {
    scanf("%d%d", &n, &m);
    
    memset(h, -1, sizeof h);//初始化邻接表
    
    while (m--) {
        int a, b;
        scanf("%d%d", &a, &b);
        add(a, b), add(b, a);
    }
    
    bool flag = true;
    //1. 遍历所有点
    for (int i = 1; i <= n; ++i) {
        //2. 如果当前点未染色
        if (!color[i]) {
            //3. dfs进行染色
            if (!dfs(i, 1)) {
                flag = false;//染色过程出现矛盾，即不是二分图
                break;
            }
        }
    }
    
    if (flag) puts("Yes");
    else puts("No");
    
    return 0;
}