알고리즘/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])
반응형