Python中常用的数据结构---二叉树的遍历

Python中常用的数据结构—二叉树的遍历

常用的数据结构有数组、链表(一对一)、栈和队列、哈希表、树(一对多)、图(多对多)等结构。
在本目录下我们将讲解，通过python语言实现常用的数据结构。

2.二叉树的遍历

二叉树的遍历分为4种，前序遍历、中序遍历、后序遍历、层序遍历
前序遍历：根节点、左节点、右节点
中序遍历：左节点、根节点、右节点
后序遍历：左节点、右节点、根节点

#二叉树的3种遍历方式
class TreeNode:
    def __init__(self, data):
        self.data = data
        self.left = None
        self.right = None

#采用递归方式构建二叉树,先序遍历创建二叉树
def create_binary_tree(input_list = []):
    if input_list is None or len(input_list) == 0:
        return None
    data = input_list.pop(0)
    if data is None:
        return None
    node = TreeNode(data)
    node.left = create_binary_tree(input_list)
    node.right = create_binary_tree(input_list)
    return node

#前序遍历
def pre_order_traversal(node):
    if node is None: return

    print(node.data)
    pre_order_traversal(node.left)
    pre_order_traversal(node.right)
    return node

#中序遍历
def in_order_traversal(node):
    if node is None: return

    pre_order_traversal(node.left)
    print(node.data)
    pre_order_traversal(node.right)
    return node


# 后序遍历
def post_order_traversal(node):
    if node is None: return

    pre_order_traversal(node.left)
    pre_order_traversal(node.right)
    print(node.data)
    return node

#测试用例
my_input_list = list([3, 2, 9, None, None, 10, None, None, 8, None, 4])
root = create_binary_tree(my_input_list)
print("前序遍历：")
pre_order_traversal(root)
print("中序遍历：")
in_order_traversal(root)
print("后序遍历：")
post_order_traversal(root)