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

문제

시간 제한메모리 제한제출정답맞힌 사람정답 비율

1 초

256 MB

72212

32054

22671

42.447%

문제

루트 없는 트리가 주어진다. 이때, 트리의 루트를 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

풀이

처음에 DFS 로 풀었다가 시간초과가 났다.

BFS 로 바꾸니 통과했다. 

 

로직은 간단한.. 1과 가까운 노드가 부모가 된다. 

1 부터 타고 올라가서 before 를 저장해두었다가 부모노드에 before 를 차례차례 넣어준다. 

 

from collections import deque

e = int(input())
parent = [0]*(e + 1)
edges = [[] for _ in range(e + 1)]
visited = [False] * (e + 1)

for _ in range(e - 1):
    start, to = map(int, input().split())
    edges[start].append(to)
    edges[to].append(start)

def bfs(start):
    q = deque([(start,0)])
    visited[start] = True

    while q:
        v,before = q.popleft()
        parent[v] = before

        for i in edges[v]:
            if not visited[i]:
                q.append((i,v))
                visited[i] = True

bfs(1)
for p in parent[2:]:
    print(p)