染色法判断二分图

题目：

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 100010,M = 200010;
int e[N],ne[N],h[N],idx;
int color[M];
int n,m;
void add(int a,int b)
{
	e[idx] = b;
	ne[idx] = h[a];
	h[a] = idx++;
}//建图 

bool dfs(int u,int x)
{
	color[u] = x;//染色 
	for(int i = h[u];i != -1;i = ne[i])//遍历当前节点周围的节点 
	{
		int j = e[i];
		if(!color[j])//如果没有染色就dfs 
		{
			dfs(j,3-x);//将2->1,1->2 
		}
		else//如果染过色，就判断是否是相同的颜色，如果是，就不是二分图 
		{
			if(color[j] == x)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])//如果没有染色就dfs染色 
		{
			if(!dfs(i,1))
			{
				flag = false ;
			}
		}
	}
	if(!flag)cout<<"No";
	else cout<<"Yes";
	return 0;
}