알고리즘/백준

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

알감자 2022. 1. 5. 23:20
 

11725번: 트리의 부모 찾기

루트 없는 트리가 주어진다. 이때, 트리의 루트를 1이라고 정했을 때, 각 노드의 부모를 구하는 프로그램을 작성하시오.

www.acmicpc.net

 

코드

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])
저작자표시 (새창열림)