글 읽기 - 왜 틀린건지 알수있을까요 34퍼에서 틀렸다고나오네요

uhs20000 3년 전 0

1. root노드 1아님 고려
2. 순서대로 나오지않는것 대비
이렇게 예외처리 다 했는데 왜 틀렸는지 잘 이해가안되네요 ㅠ
알려주실 고수분찾습니다.

import sys
sys.setrecursionlimit(1000000)
class node:
     def __init__(self,node_number=None):
        self.node_number=node_number
        self.left_node_number=-1
        self.right_node_number=-1
        self.child={}
        self.height=1
        self.parent=True
        self.lines=[]
        
def leftfirst(cur_node,lines):

    if cur_node.left_node_number!=-1:
        leftfirst(cur_node.child[cur_node.left_node_number],lines)
    lines.append([cur_node.node_number,cur_node.height])
    if cur_node.right_node_number!=-1:
        leftfirst(cur_node.child[cur_node.right_node_number],lines)

    return lines

def set_height(cur_node):
    if cur_node.left_node_number!=-1:
        cur_node.child[cur_node.left_node_number].height=cur_node.height+1
        set_height(cur_node.child[cur_node.left_node_number])
        
    if cur_node.right_node_number!=-1:
        cur_node.child[cur_node.right_node_number].height=cur_node.height+1
        set_height(cur_node.child[cur_node.right_node_number])
        
from collections import defaultdict
n=int(input())
root_check=[node(i) for i in range (n+1)]
lines=[]
for i in range(n):
    
    node_number,left_node_number,right_node_number=map(int,input().split())

    if left_node_number!=-1:
        
        root_check[node_number].child[left_node_number]=root_check[left_node_number]
        root_check[node_number].left_node_number=left_node_number
        root_check[left_node_number].parent=False
        
    if right_node_number!=1:
       
        root_check[node_number].child[right_node_number]=root_check[right_node_number]
        root_check[right_node_number].parent=False
        root_check[node_number].right_node_number=right_node_number

for i in range(1,n+1):
    if root_check[i].parent==True:
        root=i
        break
        
cur=root
set_height(root_check[root])
t=leftfirst(root_check[root],lines)


height_dict=defaultdict(list)

row_dict={value[0]:idx+1 for idx,value in enumerate(t)}

for left_node_number,height in t:
    height_dict[height].append(row_dict[left_node_number])
    
height_min_max=[]

for height,node_number_list in height_dict.items():
    height_min_max.append([height,max(node_number_list)-min(node_number_list)])

final=(max(height_min_max,key=lambda x:(x[1],-x[0])))
str_=str(final[0])+" "+str(final[1]+1)
print(str_)

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

문제

문제

출처

ICPC

왜 틀린건지 알수있을까요 34퍼에서 틀렸다고나오네요

uhs20000 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말