-
Notifications
You must be signed in to change notification settings - Fork 12
/
Copy pathis_bipartite_dfs.cpp
65 lines (52 loc) · 1.48 KB
/
is_bipartite_dfs.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
//
// 二部グラフ判定 (by DFS)
//
// reference:
// DFS (深さ優先探索) 超入門! 〜 グラフ・アルゴリズムの世界への入口 〜【後編】
// https://qiita.com/drken/items/a803d4fc4a727e02f7ba
//
// verified:
// AtCoder ARC 327 D - Good Tuple Problem
// https://atcoder.jp/contests/abc327/tasks/abc327_d
//
#include <bits/stdc++.h>
using namespace std;
// 二部グラフ判定 (color は、1: 黒色, -1: 白色, 0: 未確定)
using Graph = vector<vector<int>>;
bool dfs(const Graph &G, int v, int c, vector<int> &color) {
color[v] = c;
for (auto v2 : G[v]) {
if (color[v2] == 0) {
if (!dfs(G, v2, -c, color)) return false;
} else if (color[v2] != -c) return false;
}
return true;
}
bool is_bipartite(const Graph &G) {
int N = (int)G.size();
vector<int> color(N, 0);
for (int v = 0; v < N; ++v) {
if (color[v] != 0) continue;
if (!dfs(G, v, 1, color)) return false;
}
return true;
}
/*/////////////////////////////*/
// Examples
/*/////////////////////////////*/
void ABC_327_D() {
int N, M;
cin >> N >> M;
vector<int> A(M), B(M);
for (int i = 0; i < M; ++i) cin >> A[i], --A[i];
for (int i = 0; i < M; ++i) cin >> B[i], --B[i];
vector<vector<int>> G(N);
for (int i = 0; i < M; ++i) {
G[A[i]].push_back(B[i]);
G[B[i]].push_back(A[i]);
}
cout << (is_bipartite(G) ? "Yes" : "No") << endl;
}
int main() {
ABC_327_D();
}