1068번 - 트리

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

1068번: 트리

트리의 노드별로 부모를 입력받고, 지워야 할 노드의 번호를 입력받는다.

그런 다음 트리를 만들어주는데 이때 지워야 할 노드를 부모에 추가하는 과정은 생략한다. 왜냐하면 어차피 지워야 할 노드이기 때문이다.

 

덱을 생성하고 그 덱의 자식 노드가 있으면 while 반복문을 통해 자식 노드를 덱에 추가하며 제거해준다. 

왜냐하면 트리에서 한 노드를 제거하면 그 노드의 자식노드도 제거되기 때문이다.

해당 노드의 자식을 다 제거하면 최종적으로 지워야 할 노드를 삭제한다.

 

제거 과정을 거친 후 트리의 노드를 탐색하며, 존재하면서 자식 노드가 존재하지 않는 리프 노드를 탐색한다.

# 트리
import sys
from collections import deque
input = sys.stdin.readline

N = int(input())
parent = list(map(int, input().split()))
delete = int(input())

tree = {}
for i in range(N):
	tree[i] = deque()
for i in range(N):
    value = parent[i]
    if i == delete:
        continue
    if value != -1: # 루트 노드가 아니면
        tree[value].append(i)

Q = deque([delete])
while Q:
    cur = Q.popleft()
    while tree[cur]:
        node = tree[cur].popleft()
        Q.append(node)
    del tree[cur]

answer = 0
for i in range(N):
    if i in tree and len(tree[i]) == 0:
        answer += 1
print(answer)