글 읽기 - 틀렸다고 하는데 이유를 모르겠습니다.

japasang 3년 전 0

주어진 예제 통과하고 나름대로 예제 만들어서 해도 맞게 동작하는데,

제출하면 3퍼센트에서 틀렸다고 하는 이유를 도저히 모르겠습니다.

class Node:
    def __init__(self, data, left, right):
        self.parent = -1
        self.data = data
        self.left = left
        self.right = right
        self.level = 0
        self.x = 0

tree = dict()
def define_level(node, level): #pre_order
    node.level = level
    level += 1
    if node.left != -1:
        define_level(tree[node.left], level)
    if node.right != -1:
        define_level(tree[node.right], level)

def define_x(node): #in_order
    global x_count
    if node.left != -1:
        define_x(tree[node.left])
    node.x = x_count
    x_count += 1
    if node.right != -1:
        define_x(tree[node.right])

n = int(input())

for i in range(1, n + 1):
    tree[i] = Node(i, -1, -1)

for _ in range(n):
    data, left, right = map(int, input().split())
    tree[data].left = left
    tree[data].right = right
    if left != -1:
        tree[left].parent = data
    if right != -1:
        tree[right].parent = data
root_node = 0
for i in range(1, n+1):
    if tree[i].parent == -1:
        root_node = i
        break
        
define_level(tree[root_node], 1) #pre_order
x_count = 1
define_x(tree[root_node]) #in_order

level_dict = dict()
for key, node in tree.items():
    try:
        level_dict[node.level].append(node.data)
    except:
        level_dict[node.level] = [node.data]

max_level = 1
max_area = 0
for level, data in level_dict.items():
    area = tree[data[-1]].x - tree[data[0]].x + 1
    if max_area == area:
        max_level = min(max_level, level)
    if max_area < area:
        max_level = level
        max_area = area

print(max_level, max_area)
    

        

gmldms784 2년 전 0

root_node = 0 라고 쓰셨는데

tree의 0번 노드가 root_node가 아닐 수도 있습니다.

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

문제

문제

출처

ICPC

틀렸다고 하는데 이유를 모르겠습니다.

japasang 3년 전 0

gmldms784 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말