acwing 860. 染色法判定二分图

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

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

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

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

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

数据范围
1≤n,m≤10⁵

输入样例：
4 4
1 3
1 4
2 3
2 4
输出样例：
Yes

染色法

将所有点分成两个集合，使得所有边只出现在集合之间，就是二分图
二分图：一定不含有奇数环，奇数环分配到最后一定会有两种颜色一样挨着，所以不符合条件
但包含长度为偶数的环， 不一定是连通图

这题采用dfs （深度优先遍历）
思路

• 染色可以使用1和2区分不同颜色，用0表示未染色
• 遍历所有点，每次将未染色的点进行dfs, 默认染成1或者2
• 由于某个点染色成功不代表整个图就是二分图,因此只有某个点染色失败才能立刻break/return
  注： 染色失败相当于存在相邻的2个点染了相同的颜色

code：

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

using namespace std;

const int N = 100010, M = N * 2;

int n,m;
int h[N], e[M], ne[M], idx;
int color[N];

void add(int a, int b)  // 添加一条边a->b
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++ ;
}

bool dfs(int u, int c) // 因为color数组初始化是0， 这里染色成1或者2
{
    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)) // 如果之前染色是1， 这里就是2（1，2换）
                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, 1))
            {
                flag = false;
            }
        }
    
    if (flag) puts("Yes");  // 说明染色一路上畅通无阻
    else puts("No");
    return 0;
}