[백준] 11725번 트리의 부모 찾기 (파이썬)

11725번: 트리의 부모 찾기

 

코드

from collections import deque

n = int(input())
tree = list([] for _ in range(n+1))
check_li = list(False for _ in range(n+1))

for _ in range(n-1):
    x, y = map(int,input().split())
    tree[x].append(y)
    tree[y].append(x)

queue = deque([])
queue.append(1)
answer = {}

while queue:
    x = queue.popleft()
    for i in tree[x]:
        if not check_li[i]:
            queue.append(i)
            answer[i] = x
            check_li[i] = True

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