글 읽기 - 이진트리 구현 (파이썬)

eyeben 2년 전 0

오류내용: 틀렸습니다.

연결리스트 이진트리로 구현해서 풀어봤는데

아무리 봐도 문제점을 모르겠네요 ㅠㅠ

(search 함수는 입력받은 data의 노드를 찾아 리턴합니다)

리스트로 구현하면 더 깔끔하겠지만 이 코드는 뭐가 문제일까요

class Node():
    def __init__(self,data,left=None,right=None):
        self.data=data
        self.right=right
        self.left=left

def search(node, data):
    if node==None:
        return None
    if node.data==data:
        return node
    else:
        a=search(node.left,data)
        b=search(node.right,data)
        if a==None:
            return b
        else:
            return a

def preorder(node):
    if node==None:
        return
    print(node.data, end='')
    preorder(node.left)
    preorder(node.right)

def inorder(node):
    if node==None:
        return
    inorder(node.left)
    print(node.data, end='')
    inorder(node.right)

def postorder(node):
    if node==None:
        return
    postorder(node.left)
    postorder(node.right)
    print(node.data, end='')

N=int(input())
a,b,c=map(str,input().split())
head=Node(a,Node(b),Node(c))

for i in range(N-1):
    a,b,c=map(str,input().split())
    node=search(head, a) 
    if b!='.':
        node.left=Node(b)
    if c!='.':
        node.right=Node(c)
preorder(head)
print()
inorder(head)
print()
postorder(head)

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

문제

문제

출처

ICPC

이진트리 구현 (파이썬)

eyeben 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말