欢迎使用优快云-markdown编辑器

二叉树结构与遍历

最新推荐文章于 2021-01-07 23:01:28 发布

转载最新推荐文章于 2021-01-07 23:01:28 发布 · 170 阅读

文章标签：

#python树形结构

算法专栏收录该内容

12 篇文章

订阅专栏

本文介绍了一种基本的数据结构——二叉树，并详细解释了如何定义二叉树节点，进行左、右子节点的插入操作。此外，还探讨了三种主要的二叉树遍历方法：前序遍历、中序遍历和后序遍历。

定义树结构

class BinaryTree:
    def __init__(self, value):
        self.value = value
        self.left_child = None
        self.right_child = None

测试树结构

tree = BinaryTree('a')
print(tree.value) # a
print(tree.left_child) # None
print(tree.right_child) # None

右侧插入

def insert_right(self, value):
    if self.right_child == None:
        self.right_child = BinaryTree(value)
    else:
        new_node = BinaryTree(value)
        new_node.right_child = self.right_child
        self.right_child = new_node

左侧插入

def insert_left(self, value):
    if self.left_child == None:
        self.left_child = BinaryTree(value)
    else:
        new_node = BinaryTree(value)
        new_node.left_child = self.left_child
        self.left_child = new_node

插入节点
`a_node = BinaryTree(‘a’)
a_node.insert_left(‘b’)
a_node.insert_right(‘c’)

b_node = a_node.left_child
b_node.insert_right(‘d’)

c_node = a_node.right_child
c_node.insert_left(‘e’)
c_node.insert_right(‘f’)

d_node = b_node.right_child
e_node = c_node.left_child
f_node = c_node.right_child

print(a_node.value) # a
print(b_node.value) # b
print(c_node.value) # c
print(d_node.value) # d
print(e_node.value) # e
print(f_node.value) # f`

前序遍历

def pre_order(self):
    print(self.value)

    if self.left_child:
        self.left_child.pre_order()

    if self.right_child:
        self.right_child.pre_order()

中序遍历

def in_order(self):
    if self.left_child:
        self.left_child.in_order()

    print(self.value)

    if self.right_child:
        self.right_child.in_order()

后序遍历

def post_order(self):
    if self.left_child:
        self.left_child.post_order()

    if self.right_child:
        self.right_child.post_order()

    print(self.value)