papamelon 242. 二分圖判定(挑戰程式設計競賽)
阿新 • • 發佈:2022-06-06
地址 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; }