트리의 부모 찾기 풀이

문제 링크: https://www.acmicpc.net/problem/11725

백준 알고리즘 기초 2/2 620에서 3번째 - 11725번 트리의 부모 찾기를 풀어보았다.

 

풀이: https://my-coding-notes.tistory.com/284 를 참고해 dfs로 풀었다.

 

C++

 

Python

import sys
sys.setrecursionlimit(1000000)

def dfs(st,p):
	for i in tree[st]:
		if p[i]==0:
			p[i]=st
			dfs(i,p)

n=int(input())
tree=[[] for i in range(n+1)]
par=[0 for i in range(n+1)]
for i in range(n-1):
	a,b=map(int,input().split())
	tree[a].append(b)
	tree[b].append(a)
dfs(1,par)
for i in range(2,n+1):
	print(par[i])

Java