글 읽기 - 1068 트리 틀렸습니다. 왜. 틀렸을까요..?

colinder 3년 전 0

문제에서 주어진 cut_node의 자식 트리를 잘라 버리고

모든 leaf_node를 세어준다음 -1(cut_node)을 해주면 정답이 나올 것이라고 생각했는데

70% 정도 가다가 틀렸습니다.

왜 틀렸는지 모르겠어서, 도움 부탁 드리겠습니다.ㅜㅜㅜㅜ

import sys
from collections import deque

def BFS(i):
    q = deque([i])
    leaf_node = 0
    while q:
        x = q.popleft()
        if len(tree[x]) != 0:
            for j in tree[x]:
                q.append(j)
        else:
            leaf_node += 1
    return print(leaf_node - 1)


N = int(sys.stdin.readline().rstrip())

parents = list(map(int, sys.stdin.readline().split()))

tree = [[] for i in range(N)]
for i in range(N):
    if parents[i] == -1:
        root = i
    else:
        tree[parents[i]].append(i)
        
cut_node = int(sys.stdin.readline().rstrip())
tree[cut_node] = []

BFS(root)

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

1068 트리 틀렸습니다. 왜. 틀렸을까요..?

colinder 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말