[C++][算法基础]判定二分图(染色法)

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

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

输入格式

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

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

输出格式

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

数据范围

1≤n,m≤

输入样例：

4 4
1 3
1 4
2 3
2 4

输出样例：

Yes

代码：

 

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;

const int N = 200010;
int n,m,a,b,idx;
int StartNode[N],NextNode[N],Edgeto[N];
int color[N];

void add(int a,int b){
    Edgeto[idx] = b;
    NextNode[idx] = StartNode[a];
    StartNode[a] = idx;
    idx++;
}

int discolouration(int x,int c){
    color[x] = c;
    for(int i = StartNode[x];i != -1;i = NextNode[i]){
        int j = Edgeto[i];
        if(color[j] == 0){
            int ans = discolouration(j,3 - c);
            if(ans == 0){
                return 0;
            }
        }else if(color[j] == color[x]){
            return 0;
        }
    }
    return 1;
}

int main(){
    cin>>n>>m;
    memset(StartNode, -1, sizeof StartNode);
    while(m--){
        cin>>a>>b;
        add(a,b);
        add(b,a);
    }
    int flag = 1;
    for(int i = 1;i <= n;i ++){
        if(color[i] == 0){
            int res = discolouration(i,0);
            if(res == 0){
                flag = 0;
            }
        }
    }
    if(flag == 0){
        cout<<"No";
    }else{
        cout<<"Yes";
    }
    return 0;
}