python二叉树的建立和遍历

数的存储结构使用的是数组存储，如果是想建立完全二叉树的话只需按顺序输入格节点数值，输入’#‘结束。如果建立非完全二叉树，可以在叶子节点的值中输入’null’来表示值为空，输入’#'结束。遍历就直接对数组进行层次遍历，格局节点序号（索引值）计算节点的深度，公式是 $depth=lo{g}_{2}n$。

import math

#建立树节点结构
class TreeNode:
    value = None
    depth = None
    num = None   
    def __init__(self, value):
        self.value = value
        
#建立二叉树
def createTree():
    tree = [] #存储各节点的list
    num = 1 #记录当前节点序号
    value = input("value:") #输入节点值

    while value != "#":
        node = TreeNode(value=value) #创建节点
        node.num = num
        node.depth = int(math.log(num)/math.log(2) + 1)#计算节点深度
        tree.append(node)
        num += 1
        value = input("value:")
        
    return tree

def treeTravesal(tree):
    depth = 1
    for node in tree:
        if depth < node.depth:#判断节点所在深度，深度大的就换行并更新深度
            print()
            depth = node.depth
        print(node.value, " ", end='')

if __name__ == '__main__':
    tree = createTree()
    treeTravesal(tree)