二叉树相关代码（python）

最新推荐文章于 2024-05-01 17:47:55 发布

一只小菜鸟W

最新推荐文章于 2024-05-01 17:47:55 发布

阅读量339

点赞数 1

文章标签： python 二叉树

本文链接：https://blog.csdn.net/LWJ_1/article/details/109616568

版权

二叉树相关代码（python）

初始化一颗二叉树

在这里插入图片描述

class BiTreeNode:
    def __init__(self, data):
        self.data = data
        self.lchild = None   # 左孩子
        self.rchild = None # 右孩子

a = BiTreeNode("A")
b = BiTreeNode("B")
c = BiTreeNode("C")
d = BiTreeNode("D")
e = BiTreeNode("E")
f = BiTreeNode("F")
g = BiTreeNode("G")

e.lchild = a
e.rchild = g
a.rchild = c
c.lchild = b
c.rchild = d
g.rchild = f

root = e

前序遍历

遍历结果：EACBDGF

def pre_order(root):
    if root:
        print(root.data, end=',')
        pre_order(root.lchild)
        pre_order(root.rchild)
        
pre_order(root)
#E,A,C,B,D,G,F,

中序遍历

遍历结果：ABCDEGF

def in_order(root):
    if root:
        in_order(root.lchild)
        print(root.data, end=',')
        in_order(root.rchild)

in_order(root)
#A,B,C,D,E,G,F,

后序遍历

遍历结果：BDCAFGE

def post_order(root):
    if root:
        post_order(root.lchild)
        post_order(root.rchild)
        print(root.data, end=',')

post_order(root)
#B,D,C,A,F,G,E,

层次遍历

遍历结果：EAGCFBD

from collections import deque
def level_order(root):
    queue = deque()
    queue.append(root)
    while len(queue) > 0: # 只要队不空
        node = queue.popleft()
        print(node.data, end=',')
        if node.lchild:
            queue.append(node.lchild)
        if node.rchild:
            queue.append(node.rchild)

level_order(root)
#E,A,G,C,F,B,D,