[AcWing]860. 染色法判定二分图（C++实现）染色法判定二分图模板题

1. 题目

2. 读题（需要重点注意的东西）

思路：
首先要知道，什么是二分图。

二分图的定义如下： 二分图又称作二部图，是图论中的一种特殊模型。
设G=(V,E)是一个无向图，如果顶点V可分割为两个互不相交的子集(A,B)，并且图中的每条边（i，j）所关联的两个顶点i和j分别属于这两个不同的顶点集(i
in A,j in B)，则称图G为一个二分图。
简而言之：如果把无向图中的节点划分为两个互不相交的子集，图中一条边的两个端点分别属于这两个集合，这样的图就叫二分图。

因此，按照这个思路，我们可以选择第一个点，按dfs的思想，依次对所有的节点进行染色，如果任意一条边的两个顶点属于不同的颜色，则它是一个二分图。如下图所示。
其中，箭头表示dfs中回溯的过程。

---

代码实现思路：

3. 解法

---------------------------------------------------解法---------------------------------------------------

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

using namespace std;

const int N = 100010, M = 200010;// 无向图，两个节点间要存两条边

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; // 记录当前点的颜色为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; // 3-c表示：如果染过1号颜色（c = 1），则染2号颜色（3-c = 2）；如果染过2号颜色，则染1号颜色
        }
        else if (color[j] == c) 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;
    for (int i = 1; i <= n; i ++ )
        if (!color[i]) // 如果当前节点未染过色，用dfs去染色
        {
            if (!dfs(i, 1)) // 如果有矛盾发生（dfs为false） （1传入dfs，表示将第i个点染成1代表的颜色）
            {
                flag = false;
                break;// 如果有矛盾，直接break，返回No即可
            }
        }

    if (flag) puts("Yes");
    else puts("No");

    return 0;
}

注意：
在dfs中，只有当染色情况出现冲突时，才会返回false，即

else if (color[j] == c) return false; // 如果当前节点已经染过颜色，则判断是否有矛盾即可

因此dfs的功能是： 如果当前节点没有被染色，染上合法的颜色；如果当前节点已经被染色，判断该颜色是否合法，如果不合法，返回false，不需要再继续染色了。

4. 可能有帮助的前置习题

5. 所用到的数据结构与算法思想

• dfs
• 染色法判断二分图
• 邻接表存储稀疏图

6. 总结

染色法判断二分图算法的模板题，背下来！