[백준] 5639번 : 이진 검색 트리

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

5639번: 이진 검색 트리

Node의 자료구조를 만들어 문제를 해결 하였습니다. InsertNode함수는 현재 루트 노드로부터 해당 노드를 인서트 합니다.

재귀 함수를 사용하여 preOrder 함수는 전위순회로 출력하고 postOrder 함수는 후위순회로 출력합니다.

import sys
sys.setrecursionlimit(10**6)

class Node(object):
    def __init__(self,data=None):
        self.data = data
        self.leftnode = None
        self.rightnode = None
    
        
    def insertNode(self, value):
        currentNode = self
        while True:
            if currentNode.data > value:
                if not currentNode.leftnode:
                    currentNode.leftnode = Node(value)
                    break
                else:
                    currentNode = currentNode.leftnode
            else:
                if not currentNode.rightnode:
                    currentNode.rightnode = Node(value)
                    break
                else:
                    currentNode = currentNode.rightnode
            

currentNode = Node(int(input()))

while True:
    try:
        currentNode.insertNode(int(input()))
    except:
        break
    

def postOrder(node):
    if node is None:
        return
    
    postOrder(node.leftnode)
    postOrder(node.rightnode)
    print(node.data)

def preOrder(node):
    if node is None:
        return
    
    print(node.data)
    preOrder(node.leftnode)
    preOrder(node.rightnode)
    
postOrder(currentNode)