860. 染色法判定二分图

题目

给定一个 n 个点 m 条边的无向图，图中可能存在重边和自环。

请你判断这个图是否是二分图。

输入格式
第一行包含两个整数 n 和 m。

接下来 m 行，每行包含两个整数 u 和 v，表示点 u 和点 v 之间存在一条边。

输出格式
如果给定图是二分图，则输出 Yes，否则输出 No。

数据范围
1≤n,m≤105
输入样例：
4 4
1 3
1 4
2 3
2 4
输出样例：
Yes

思路：

遍历图中的所有点，对各个连通块中的某个点进行dfs深搜染色，并判断连通块染色是否有矛盾

代码

#include <iostream>
#include <cstring>

using namespace std;
const int N = 100010, M = 2 * N;
int h[N], e[M], ne[M], idx;
int color[N];//标记当前点的染色值，0表示没有染色，1表示白色，2表示黑色
int n, m;

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

// 深搜，如果染色过程中出现矛盾，则返回false,否则返回true
bool dfs(int u, int c){
    // 把当前节点染成c
    color[u] = c;
    
    // 遍历当前节点的所有相邻节点，如果当前节点是1，则把相邻点染成2,如果当前节点是2，则把相邻点染成1
    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);
    }
    
    int flag = true;//对图染色的过程中，是否有矛盾的标记
    // 遍历图中的所有点，对各个连通块中的某个点进行深搜染色，并判断连通块染色是否有矛盾
    for(int i = 1; i <= n; i++){
        if(!color[i]){
            if(!dfs(i, 1)){//默认对未处理的连通块的该点染成1号颜色
                flag = false;
                break;
            }
        }
    }
    
    if(flag) puts("Yes");
    else puts("No");
    
    return 0;
}