acwing算法基础课：二分图算法（染色法判别二分图算法）

染色法判别二分图模板

时间复杂度是 O(n+m), n 表示点数，m表示边数

int n;      // n表示点数
int h[N], e[M], ne[M], idx;     // 邻接表存储图
int color[N];       // 表示每个点的颜色，-1表示未染色，0表示白色，1表示黑色

// 参数：u表示当前节点，c表示当前点的颜色
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)
        {
            if (!dfs(j, !c)) return false;
        }
        else if (color[j] == c) return false;
    }

    return true;
}

bool check()
{
    memset(color, -1, sizeof color);
    bool flag = true;
    for (int i = 1; i <= n; i ++ )
        if (color[i] == -1)
            if (!dfs(i, 0))
            {
                flag = false;
                break;
            }
    return flag;
}

例题

给定一个n个点m条边的无向图，图中可能存在重边和自环。请你判断这个图是否是二分图。

#include <iostream>
#include <cstring>

using namespace std;

const int N = 100010;

int n, m;
int h[N], e[N], ne[N], 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])
		{
			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, 1))
			{
				flag = false;
				break;
			}
		}
	}
	
	if(!flag) cout << "No" << endl;
	else cout << "Yes" << endl;
	
	return 0;
}

测试样例

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