트리의 부모 찾기 성공
시간 제한메모리 제한제출정답맞힌 사람정답 비율
1 초 | 256 MB | 37330 | 16119 | 11689 | 42.273% |
문제
루트 없는 트리가 주어진다. 이때, 트리의 루트를 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
import sys
from collections import deque
sys.setrecursionlimit(10 ** 7)
input = sys.stdin.readline
if __name__ == '__main__':
N = int(input())
Tree = [[] for i in range(N+1)]
Parent = [0 for _ in range(N+1)]
for _ in range(N-1):
x, y = list(map(int, input().split()))
Tree[x].append(y)
Tree[y].append(x)
def dfs(start):
for i in Tree[start]:
if Parent[i] == 0:
Parent[i] = start
dfs(i)
dfs(1)
for i in range(2, N+1):
print(Parent[i])
댓글