程序员面试算法宝典-3.3 如何从顶部开始逐层打印二叉树结点数据

题目描述：

如何从顶部开始逐层打印二叉树结点数据。

分析与解答：

采用队列来存储当前遍历到的结点的孩子结点，从而实现二叉树的层序遍历。

from collections import deque

class BiTNode:
    def __init__(self):
        self.data = None
        self.lchild = None
        self.rchild = None

# 方法功能：把有序数组转换为二叉树
def arraytotree(arr, start, end):
    root = None
    if end>=start:
        root = BiTNode()
        mid = (start+end+1)//2
        # 树的根节点为数组中间的元素
        root.data = arr[mid]
        # 递归的用左半部分数组构造root左子树
        root.lchild = arraytotree(arr, start, mid-1)
        # 递归的用右半部分数组构造root的右子树
        root.rchild = arraytotree(arr, mid+1, end)
    else:
        root = None
    return root

"""
方法一：空间复杂度O(N)
方法功能：用层序遍历的方式打印出二叉树结点的内容
输入参数：root：二叉树根节点
"""
def printTreeLayer(root):
    if root == None:
        return
    queue = deque()
    # 树根节点进队列
    queue.append(root)
    while len(queue)>0:
        p = queue.popleft()
        # 访问当前结点
        print(p.data)
        # 如果这个结点的左孩子不为空则入队列
        if p.lchild != None:
            queue.append(p.lchild)
        # 如果这个结点的右孩子不为空则入队列
        if p.rchild != None:
            queue.append(p.rchild)

if __name__=="__main__":
    arr = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
    root = arraytotree(arr, 0, len(arr)-1)
    print("树的层序遍历结果为：")
    printTreeLayer(root)