글 읽기 - 5639번 파이썬

qawesdxc 4년 전 0

17~18%에서 시간초과가 나네요.

질문게시판들을 보면 많은 분들이 파이썬으로 풀게되면 비슷한 구간에서 오류가 난다는데

혹시 이유를 알 수 있난 방법이 있나요??

import sys

class Node:
    def __init__(self, data):
        self.data = data
        self.lchild = None
        self.rchild = None
        self.visit = False

class make_tree:
    def __init__(self):
        self.root = None

    def making(self, data):
        if(self.root == None):
            self.root = Node(data)

        else:
            current = self.root
            while(True):
                if(current.data > data):
                    if(current.lchild == None):
                        current.lchild = Node(data)
                        break
                    current = current.lchild

                if(current.data < data):
                    if(current.rchild == None):
                        current.rchild = Node(data)
                        break
                    current = current.rchild

def DFS(node):
    global prenum

    if(node.visit == True):
        return

    if(node.lchild == None and node.rchild == None):
        return

    node.visit = True

    if (node.lchild != None and node.lchild.visit == False):
        prenum += 1
        DFS(node.lchild)
        prenum += 1 #postnum
        result[node.lchild.data] = prenum #postnum

    if(node.rchild != None and node.rchild.visit == False):
        prenum += 1
        DFS(node.rchild)
        prenum += 1 #postnum
        result[node.rchild.data] = prenum #postnum

obj = make_tree()
prenum = 1
result = {}

while(True):
    try:
        obj.making(int(sys.stdin.readline()))
    except:
        break

DFS(obj.root)
result[obj.root.data] = 20001

for k in result.keys():
    print(k)

grasshoppertrainer 3년 전 0

재귀 함수의 깊이 때문인것 같습니다.

sys.setrecursionlimit(20_000)로 시도해보세요.

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

문제

문제

출처

ICPC

5639번 파이썬

qawesdxc 4년 전 0

grasshoppertrainer 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말