二叉树的中序遍历和后序遍历

class Node:
	def __init__(self, val):
		self.val = val
		self.left, self.right = None, None
class Tree:
	def create_tree(self, pre, mid):
		if pre:
			root_val = pre[0]
			root_val_index_mid = mid.index(root_val)
			root = Node(root_val)
			root.left = self.create_tree(pre[1:root_val_index_mid+1], mid[:root_val_index_mid])
			root.right = self.create_tree(pre[root_val_index_mid+1:], mid[root_val_index_mid+1:])
		else:
			return None
		return root

	def print_tree(self, root):
		nodes = [root]
		while nodes:
			node = nodes.pop(0)
			print(node.val)
			if node.left:
				nodes.append(node.left)
			if node.right:
				nodes.append(node.right)

	def mid_series(self, root):
		'''中序遍历'''
		stack, stack2 = [], []
		node = root
		while node or stack:
			if node:
				stack.append(node)
				node = node.left
			else:
				node = stack.pop()
				stack2.append(node.val)
				node = node.right
		print(stack2)

	def back_series(self, root):
		'''后序遍历'''
		stack, stack2 = [root], []
		while stack:
			node = stack.pop()
			stack2.insert(0, node.val)
			if node.left:
				stack.append(node.left)
			if node.right:
				stack.append(node.right)

		print(stack2)


tree = Tree()
root = tree.create_tree(['1', '2', '4', '3', '5', '6'], ['2', '4', '1', '5', '3', '6'])
# tree.print_tree(root)
tree.mid_series(root)
tree.back_series(root)

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值