BOJ 11725번: 트리의 부모 찾기

www.acmicpc.net/problem/11725

11725번: 트리의 부모 찾기

아이디어:

그냥 시작하는 부분을 1로 하고 DFS 탐색을 하면서 부모노드를 저장해주면 되는 간단한 문제였다.

import sys

N = int(sys.stdin.readline())
graph = [[] for _ in range(N+1)]

for _ in range(N-1):
    a, b = map(int, sys.stdin.readline().split())
    graph[a].append(b)
    graph[b].append(a)
    
stack = []
stack.append(1)
visit = [False]*(N+1)
visit[1] = True
ans = [0]*(N+1)

while (stack):
    n = stack.pop()
    for i in graph[n]:
        if (visit[i]==False):
            visit[i] = True
            stack.append(i)
            ans[i] = n
            
for i in range(2,N+1):
    print(ans[i])