취준시절/백준

[백준 11725] 트리의 부모 찾기 - Python

MAYMIN 2021. 7. 1. 20:59
728x90
SMALL
n = int(input())

answer = [[] for _ in range(n+1)]

def dfs(startNode):
    #방문 할 리스트
    stack = list()
    stack.append(startNode)

    while stack:
        node = stack.pop()
        for i in tree[node]:
            answer[i].append(node)
            stack.append(i)
            tree[i].remove(node)

tree = [[] * (n+1) for _ in range(n+1)]

# 그래프 그린것
for i in range(n-1):
    node1,node2 = map(int,input().split())
    tree[node1].append(node2)
    tree[node2].append(node1)

#트리의 루트를 1이라고 하자.
dfs(1)
for i in range(2,n+1):
    print(answer[i][0])
728x90
LIST