[Python] BOJ 11725번. 트리의 부모 찾기

작성일 | In |

11725번. 트리의 부모 찾기

문제 링크

풀이 코드

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
# 11725번. 트리의 부모 찾기


from collections import deque

n = int(input())
nList = [[] for i in range(n+1)]
for i in range(n-1):
    a, b = map(int, input().split())
    nList[a].append(b)
    nList[b].append(a)

q = deque([1])
visited = [False for i in range(n+1)]
ans = {}
for i in range(1, n + 1):
    ans[i] = []

while q:
    parent = q.popleft()
    for child in nList[parent]:  # 이 노드의 자식들
        if not visited[child]:
            ans[child] = parent  # 부모를 저장해줌
            q.append(child)
            visited[child] = True

for i in range(2, n + 1):
    print(ans[i])

비고