860. 染色法判定二分图

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

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

输入格式

第一行包含两个整数 nn 和 mm。

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

输出格式

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

数据范围

1≤n,m≤1051≤n,m≤105

输入样例：

4 4
1 3
1 4
2 3
2 4

输出样例：

Yes

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 100010,M=200010;
int color[N];
int h[N],e[M],ne[M],idx;
int n,m;
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()
{
    memset(h, -1, sizeof h);
    cin>>n>>m;
    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)puts("Yes");
    else puts("No");
    return 0;
}