[백준] 1991번: 트리순회 파이썬

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

1991번: 트리 순회

굿

 

재귀함수돌때 언제 출력해야하는지만 유의하면 쉬운 문제!

풀이

 

전위 순회는 출력하고 왼오

중위 순회는 왼 출력하고 오

후위 순회는 왼오하고 출력

 

트리는 그냥 딕셔너리로 저장..

def preorder(x):
  print(x, end="")
  if tree[x][0] != ".":
    preorder(tree[x][0])
  if tree[x][1] != ".":
    preorder(tree[x][1])

def inorder(x):
  if tree[x][0] != ".":
    inorder(tree[x][0])
  print(x, end="")
  if tree[x][1] != ".":
    inorder(tree[x][1])

def postorder(x):
  if tree[x][0] != ".":
    postorder(tree[x][0])
  if tree[x][1] != ".":
    postorder(tree[x][1])
  print(x, end="")

n = int(input())
tree = {}
for i in range(n):
  data, left, right = input().split()
  tree[data] = [left, right]

preorder("A")
print()
inorder("A")
print()
postorder("A")