From 69073a4b7e9604fe7e4ce6c3925fc1ec7eb9c0eb Mon Sep 17 00:00:00 2001 From: yenru0 Date: Mon, 20 May 2024 13:58:07 +0900 Subject: [PATCH] complete 1707.py --- zeta_python/completed/1707.py | 42 +++++++++++++++++++++++++++++++++++ 1 file changed, 42 insertions(+) create mode 100644 zeta_python/completed/1707.py diff --git a/zeta_python/completed/1707.py b/zeta_python/completed/1707.py new file mode 100644 index 0000000..7f4a192 --- /dev/null +++ b/zeta_python/completed/1707.py @@ -0,0 +1,42 @@ +import sys + +input = sys.stdin.readline + + +def color_reversal(c: int) -> int: + if c == 1: + return 2 + else: + return 1 + + +def solve(V: int, E) -> str: + + D = list(range(1, V + 1)) + colored = [0] * (V + 1) + while D: + now = D.pop() + if not colored[now]: + colored[now] = 1 + for target in E[now]: + if colored[target]: + if color_reversal(colored[now]) == colored[target]: + continue + else: + return "NO" + else: + colored[target] = color_reversal(colored[now]) + D.append(target) + return "YES" + + +if __name__ == "__main__": + K = int(input()) + for _ in range(K): + V, C = map(int, input().split()) + E = [[] for _ in range(V + 1)] + for _ in range(C): + u, v = map(int, input().split()) + E[u].append(v) + E[v].append(u) + print(solve(V, E))