数的存储结构使用的是数组存储,如果是想建立完全二叉树的话只需按顺序输入格节点数值,输入’#‘结束。如果建立非完全二叉树,可以在叶子节点的值中输入’null’来表示值为空,输入’#'结束。遍历就直接对数组进行层次遍历,格局节点序号(索引值)计算节点的深度,公式是 d e p t h = l o g 2 n depth=log_{2}n depth=log2n。
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)