Problem Solving/Tree

[트리] 백준 11725번: 트리의 부모 찾기 / 실버 2

by ggyongi 2021. 6. 14.

https://www.acmicpc.net/problem/11725

11725번: 트리의 부모 찾기

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

www.acmicpc.net

import collections
import sys
sys.setrecursionlimit(100000)
n = int(input())
edges = collections.defaultdict(list)
for i in range(n-1):
    a, b = map(int, input().split())
    edges[a].append(b)
    edges[b].append(a)

answer = [0]*(n+1)
visited = [False]*(n+1)

def search(node):
    for elem in edges[node]:
        if not visited[elem]:
            visited[elem] = True
            answer[elem] = node
            search(elem)

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