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

16 May 2023 in Study on Algorithm

백준11725 트리의 부모 찾기 링크

문제

예제 입력

---

코드

import sys
from collections import deque

n = int(input())
arr = [[] for _ in range(n + 1)]
ans = [0 for _ in range(n + 1)]

for _ in range(n - 1):
    a, b = map(int, sys.stdin.readline().split())
    arr[a].append(b)
    arr[b].append(a)

q = deque()
def BFS():
    q.append(1)
    while q:
        num = q.popleft()
        for i in arr[num]:
            if ans[i] == 0:
                ans[i] = num
                q.append(i)
        
BFS()
for i in ans[2:]:
    print(i)

설명

파이썬을 통해서 사용자로부터 입력받아 BFS알고리즘을 사용하여 트리의 부모 찾기를 구현했습니다.

---

결과