[BOJ] 백준 11725 트리의 부모 찾기

태환·2024년 2월 11일
0

Coding Test

목록 보기
75/151
post-custom-banner

📌 [BOJ] 백준 11725 트리의 부모 찾기

📖 문제

📖 예제

📖 풀이

import sys
input = sys.stdin.readline
sys.setrecursionlimit(10000000)
N = int(input())

graph = [[] for _ in range(N+1)]
for _ in range(N-1):
  a, b = map(int, input().split())
  graph[a] += [b]
  graph[b] += [a]

visited = [0] * (N+1)

def DFS(start):
  for i in graph[start]:
    if visited[i] == 0:
      visited[i] = start
      DFS(i)

for i in range(1, N+1):
  if visited[i] == 0:
    DFS(i)

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

DFS/BFS를 통해 각 리스트에 인접한 노드를 불러와 해당 노드의 자식 노드로 지정해준다.

profile
연세대학교 컴퓨터과학과 석사 과정
post-custom-banner

0개의 댓글