1. 程式人生 > 其它 >papamelon 242. 二分圖判定(挑戰程式設計競賽)

papamelon 242. 二分圖判定(挑戰程式設計競賽)

地址 https://www.papamelon.com/problem/242

解答
如果圖中沒有出現奇數環,就不會出現塗色衝突.

我們使用DFS依次遍歷圖中的點將其更迭的染成1或者2類顏色,最後看所有點是否會有衝突,即可判斷是不是二分圖。

#include <iostream>
#include <vector>
#include <memory.h>


using  namespace std;

const int N = 1010;

vector<int> gra[N];
int color[N];
int n;


bool  dfs(int curr, int fill) {
	//已經塗色 有衝突
	if (color[curr] != 0 && color[curr] != fill) return false;
	//已經塗色 且無衝突
	if (color[curr] == fill) return true;

	//塗色
	color[curr] = fill;

	for (int i = 0; i < gra[curr].size(); i++) {
		int next = gra[curr][i];
		//換個顏色 塗色下一個點
		if (false == dfs(next, 3 - fill)) return false;
	}


	return true;
}

int main()
{
	scanf("%d", &n);
		memset(color, 0, sizeof color);
		for (int i = 0; i < N; i++) { gra[i].clear(); }
		
		for (int i = 0; i < n; i++) {
			int a, b; scanf("%d%d",&a,&b);
			gra[a].push_back(b);
			gra[b].push_back(a);
		}
		int flag = 1;
		
		//逐個的更換顏色  dfs 塗色每個點
		//考慮到可能不是連通,每個點都要作為dfs起點檢測
		for (int i = 0; i < n; i++) {
			if ( false == dfs(i, 1)) {
				flag = 0; break;
			}
		}

		if (flag == 1) {
			printf("Yes\n");
		}
		else {
			printf("No\n");
		}
	


	return 0;
}

我的視訊題解空間