알고리즘/BFS&DFS

백준 11725 트리의 부모 찾기

mimi04 2024. 1. 28. 18:17
import sys
sys.setrecursionlimit(10**6)
input = sys.stdin.readline

n = int(input())
graph = [[] for _ in range(n + 1)]
for _ in range(n - 1):
     a, b = map(int, input().split())
     graph[a].append(b)
     graph[b].append(a)
visited = [0] * (n + 1)   #부모 노드를 담을 리스트

def dfs(root):
     for i in graph[root]:
          if visited[i] == 0:
               visited[i] = root
               dfs(i)
dfs(1)

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

 

반응형