860. 染色法判定二分图

// 860. 染色法判定二分图.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。
//
/*
https://www.acwing.com/problem/content/description/862/

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

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

输入格式
第一行包含两个整数 n 和 m。

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

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

数据范围
1≤n,m≤105
输入样例：
4 4
1 3
1 4
2 3
2 4
输出样例：
Yes
*/


#include <iostream>
#include <cstring>

using namespace std;


const int N = 100010,M=200010;
int h[N], e[M], ne[M], idx;
int n, m;
int color[N]; int cnt;

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


bool dfs(int p, int c) {
	if (color[p] != 0 && color[p] != c) return false;
	if (color[p] == c) return true;

	color[p] = c; cnt++;
	for (int i = h[p]; i != -1; i = ne[i]) {
		int j = e[i];
		if ( !dfs(j, 3 - c)) {
			return false;
		}
	}

	return true;
}


int main()
{
	cin >> n >> m;
	memset(h, -1, sizeof h);
	for (int i = 0; i < m; i++) {
		int a, b; cin >> a >> b;
		add(a, b); add(b, a);
	}

	for (int i = 1; i <= n; i++) {
		if (color[i] == 0) {
			if (!dfs(i, 1)) {
				cout << "No" << endl;
				return 0;
			}
		}
	}

	if (cnt == n) {
		cout << "Yes" << endl;
	}
	else {
		cout << "No" << endl;
	}
	return 0;
}