染色法判定二分图

107 阅读3分钟

染色法判定二分图

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

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

输入格式

第一行包含两个整数 n𝑛 和 m𝑚。

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

输出格式

如果给定图是二分图,则输出 Yes,否则输出 No

数据范围

1≤𝑛,𝑚≤10^5^

输入样例:

4 4
1 3
1 4
2 3
2 4

输出样例:

Yes

什么叫二分图

  • 有两顶点集且图中每条边的的两个顶点分别位于两个顶点集中,每个顶点集中没有边直接相连接!

  • 说人话的定义:图中点通过移动能分成左右两部分,左侧的点只和右侧的点相连,右侧的点只和左侧的点相连。

  • 下图就是个二分图:

  • 下图不是个二分图:

如果判断一个图是不是二分图?

  • 开始对任意一未染色的顶点染色。

  • 判断其相邻的顶点中,若未染色则将其染上和相邻顶点不同的颜色。

  • 若已经染色且颜色和相邻顶点的颜色相同则说明不是二分图,若颜色不同则继续判断。

染色法

  • 将所有点分成两个集合,使得所有边只出现在集合之间,就是二分图

  • 二分图:一定不含有奇数环,可能包含长度为偶数的环, 不一定是连通图

dfs版本

  • 代码思路:
    • 染色可以使用1和2区分不同颜色,用0表示未染色
    • 遍历所有点,每次将未染色的点进行dfs, 默认染成1或者2
    • 由于某个点染色成功不代表整个图就是二分图,因此只有某个点染色失败才能立刻break/return
      • 染色失败相当于存在相邻的2个点染了相同的颜色
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int N = 1e5 + 10, M = 2e5 + 10; // 由于是无向图, 顶点数最大是N,那么边数M最大是顶点数的2倍
int e[M], ne[M], h[N], idx;
int st[N];

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

bool dfs(int u, int color) {
    st[u] = color;

    for(int i = h[u]; i != -1; i = ne[i]){
        int j = e[i];
        if(!st[j]) {
            if(!dfs(j, 3 - color)) return false;
        }else if(st[j] == color) return false;
    }
    
    return true;

}

int main(){
    int n, m;
    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);  // 无向图,a->b, b->a
    }
    
    bool flag = true;
    for(int i = 1; i <= n; i ++){
        if(!st[i]){
            if(!dfs(i, 1)){
                flag = false;
                break;
            }
        }
    }
    
    if(flag) puts("Yes");
    else puts("No");
    return 0;

}

bfs版本

  • 代码思路
    • 颜色 1 和 2 表示不同颜色, 0 表示 未染色
    • 定义queue是存PII,表示 <点编号, 颜色>,
    • 同理,遍历所有点, 将未染色的点都进行bfs
    • 队列初始化将第i个点入队, 默认颜色可以是1或2
      • while (队列不空)
      • 每次获取队头t, 并遍历队头t的所有邻边
        • 若邻边的点未染色则染上与队头t相反的颜色,并添加到队列
        • 若邻边的点已经染色且与队头t的颜色相同, 则返回false

C++ 代码

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

using namespace std;
const int N = 1e5 + 10, M = 2e5 + 10;
typedef pair<int, int> PII;

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

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

bool bfs(int u){
    int hh = 0, tt = 0;
    PII q[N];
    q[0] = {u, 1};
    st[u] = 1;

    while(hh <= tt){
        auto t = q[hh ++];
        int ver = t.first, c = t.second;
    
        for (int i = h[ver]; i != -1; i = ne[i]){
            int j = e[i];
    
            if(!st[j])
            {
                st[j] = 3 - c;
                q[++ tt] = {j, 3 - c};
            }
            else if(st[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);
    }
    
    int flag = true;
    for(int i = 1; i <= n; i ++) {
        if (!st[i]){
            if(!bfs(i)){
                flag = false;
                break;
            }
        }
    }
    
    if (flag) puts("Yes");
    else puts("No");
    return 0;

}