백준 11725번 트리의 부모 찾기

문제 링크: https://www.acmicpc.net/problem/11725

Author: skysign

백준 11725번 트리의 부모 찾기/main.py

@@ -0,0 +1,36 @@
+import sys
+from typing import List, Deque
+from collections import deque
+
+
+def solve():
+    N = int(sys.stdin.readline().strip())
+    nodes: List[List[int]] = [[] for n in range(N + 1)]
+    parents: List[int] = [0 for _ in range(N + 1)]
+    parents[1] = 1
+
+    for _ in range(N - 1):
+        fr, to = list(map(int, sys.stdin.readline().strip().split(' ')))
+        nodes[fr].append(to)
+        nodes[to].append(fr)
+
+    bfs(1, nodes, parents)
+
+    for idx in range(2, N + 1):
+        print(parents[idx])
+
+
+def bfs(p, nodes, parents):
+    queue: Deque = deque([p])
+
+    while queue:
+        p = queue.popleft()
+
+        for child in nodes[p]:
+            if parents[child] == 0:
+                parents[child] = p
+                queue.append(child)
+
+
+if __name__ == '__main__':
+    solve()

백준 11725번 트리의 부모 찾기/test1.txt

@@ -0,0 +1,7 @@
+7
+1 6
+6 3
+3 5
+4 1
+2 4
+4 7

백준 11725번 트리의 부모 찾기/test1_answer.txt

@@ -0,0 +1,6 @@
+4
+6
+1
+3
+1
+4

백준 11725번 트리의 부모 찾기/test2.txt

@@ -0,0 +1,12 @@
+12
+1 2
+1 3
+2 4
+3 5
+3 6
+4 7
+4 8
+5 9
+5 10
+6 11
+6 12

백준 11725번 트리의 부모 찾기/test2_answer.txt

@@ -0,0 +1,11 @@
+1
+1
+2
+3
+3
+4
+4
+5
+5
+6
+6

백준 11725번 트리의 부모 찾기/test_main.py

@@ -0,0 +1,26 @@
+import sys
+from pathlib import Path
+from unittest import TestCase
+from main import solve
+
+
+class Test(TestCase):
+    def my_solve(self, testcase_input):
+        sys.stdin = open(testcase_input, 'r')
+        stdout = sys.stdout
+        sys.stdout = open('stdout.txt', 'w')
+        solve()
+        sys.stdout.close()
+        sys.stdout = stdout
+
+    def test_solve(self, testcase_number: str):
+        self.my_solve('test' + testcase_number + '.txt')
+        self.assertEqual(
+            Path('test' + testcase_number + '_answer.txt').read_text().strip(),
+            Path('stdout.txt').read_text().strip())
+
+    def test1_solve(self):
+        self.test_solve('1')
+
+    def test2_solve(self):
+        self.test_solve('2')