[Baekjoon] 11725. 트리의 부모 찾기

문제 설명

문제

루트 없는 트리가 주어진다. 이때, 트리의 루트를 1이라고 정했을 때, 각 노드의 부모를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 노드의 개수 N (2 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N-1개의 줄에 트리 상에서 연결된 두 정점이 주어진다.

출력

첫째 줄부터 N-1개의 줄에 각 노드의 부모 노드 번호를 2번 노드부터 순서대로 출력한다.

예제 입력 1

7
1 6
6 3
3 5
4 1
2 4
4 7

예제 출력 1

4
6
1
3
1
4

예제 입력 2

12
1 2
1 3
2 4
3 5
3 6
4 7
4 8
5 9
5 10
6 11
6 12

예제 출력 2

1
1
2
3
3
4
4
5
5
6
6

출처

• 문제를 만든 사람: baekjoon
• 잘못된 조건을 찾은 사람: jh05013

알고리즘 분류

• 그래프 이론
• 그래프 탐색
• 트리
• 너비 우선 탐색
• 깊이 우선 탐색

문제 풀이

# Graph # Tree # DFSBFS

Graph 자료구조와 DFS/BFS 알고리즘을 사용하는 Tree 문제입니다.

풀이 과정

루트가 1로 주어진 임의의 트리에서 각 노드의 부모 노드를 찾는 문제입니다.

루트 노드에서 출발해, 연결된 노드들(부모 노드 제외)의 부모를 현재 노드로 지정해주면 되는 간단한 문제입니다.

전체 코드

import sys
input = sys.stdin.readline
from collections import deque

def bfs(root):
    dp = [-1 for _ in range(N+1)] # 부모 노드
    q = deque()
    q.append(root)
    dp[root] = 0
    while q:
        cv = q.popleft()
        for nv in tree[cv]:
            if dp[nv] != -1: continue
            dp[nv] = cv
            q.append(nv)
    return dp
# main
N = int(input())
tree = [[] for _ in range(N+1)]
for _ in range(N-1):
    a,b = map(int,input().split())
    tree[a].append(b)
    tree[b].append(a) 
print(*bfs(1)[2:],sep='\n')

배운 점