알고리즘 문제 풀이

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

김혠 2024. 10. 14. 11:00

문제링크

 

최종코드(코드링크)

import sys
from sys import setrecursionlimit
from collections import defaultdict

setrecursionlimit(10**9)
# input = sys.stdin.readline
N = int(input())

visit = [False] * (N + 1)
tree = defaultdict(list)
for _ in range(N - 1):
    a, b = map(int, input().split())
    tree[a].append(b)
    tree[b].append(a)

parents = [0] * (N + 1)
def find(node):
    visit[node] = True
    node_list = tree[node]
    for child in node_list:
        if not visit[child]:
            parents[child] = node
            find(child)
find(1)
print('\n'.join(map(str, parents[2:])))

아무리 제출해도 런타임 에러가 발생하였는데..

setrecursionlimit(10**9)를 작성하여 해결했습니다 !