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

강재원·2022년 12월 1일
0

[코딩테스트] Python

목록 보기
192/200
post-custom-banner



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

import sys
sys.setrecursionlimit(10**6)

n=int(input())
arr=[[] for _ in range(n+1)]
num=[0]*(n+1)
check=[False]*(n+1)

for _ in range(n-1):
    u,v=map(int,input().split())
    arr[u].append(v)
    arr[v].append(u)

bf=1

def dfs(a):
    global bf
    check[a]=True
    num[a]=bf
    for i in arr[a]:
        if check[i]==False:
            bf=a
            dfs(i)

dfs(1)

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


profile
개념정리 & 문법 정리 & 알고리즘 공부
post-custom-banner

0개의 댓글