[백준] 11725번 문제 (트리의 부모 찾기) 파이썬(Python) 풀이

https://www.acmicpc.net/problem/11725

11725번: 트리의 부모 찾기

import sys
from collections import deque

input = sys.stdin.readline

n = int(input())
tree = [[] for _ in range(n + 1)]
parent = [0 for _ in range(n + 1)]

for _ in range(n - 1):
    a, b = map(int, input().split())
    tree[a].append(b)
    tree[b].append(a)

q = deque()
q.append(1)
while q:
    node = q.popleft()
    for i in tree[node]:
        if parent[i] == 0:
            parent[i] = node
            q.append(i)

for i in parent[2:]:
    print(i)

트리의 루트가 1이라고 했으니 1부터 큐에 넣어주고 bfs를 통해 탐색을 하면 해당 노드의 부모 노드를 찾을 수 있다.

parent 배열로 해당 노드의 부모 노드를 표시해주고, 출력 조건이 2번 노드부터라고 했으니,

parent[2:]로 슬라이싱된 리스트에서 for문으로 출력을해줬다.