二叉树高度的代码解析_剑指offer 从上到下打印二叉树

题目描述

从上往下打印出二叉树的每个节点，同层节点从左至右打印。

解析：这实际上就是二叉树的层序遍历。规律如下：每次打印节点，如果该节点有子节点，则将子节点放入队列末尾，然后取出队列头部最早进入队列的节点，重复上述操作直至队列为空。用list模拟队列。

代码如下：

class Solution:
    # 返回从上到下每个节点值列表，例：[1,2,3]
    def PrintFromTopToBottom(self, root):
        # write code here
        res = []
        queue = []
        if not root:
            return res
        queue.append(root)
        while queue:
            tmp = queue.pop(0)
            res.append(tmp.val)
            if tmp.left:
                queue.append(tmp.left)
            if tmp.right:
                queue.append(tmp.right)
        return res

扩展1：分行从上到下打印二叉树

题目描述

从上到下按层打印二叉树，同一层结点从左至右输出。每一层输出一行。

解析：层序遍历的变种，本质相同，方法类似。

代码如下：

class Solution:
    # 返回二维列表[[1,2],[4,5]]
    def Print(self, pRoot):
        # write code here
        res = []
        if not pRoot:
            return res
        queue = [pRoot]
        while queue:
            level = []
            next = []
            for tmp in queue:
                level.append(tmp.val)
                if tmp.left:
                    next.append(tmp.left)
                if tmp.right:
                    next.append(tmp.right)
            res.append(level)
            queue = next[:]
        return res

扩展2：之字形打印二叉树

题目描述

请实现一个函数按照之字形打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右至左的顺序打印，第三行按照从左到右的顺序打印，其他行以此类推。

解析：层序遍历的变种。

代码如下：

class Solution:
    def Print(self, pRoot):
        # write code here
        res = []
        if not pRoot:
            return res
        queue = [pRoot]
        k = 0
        while queue:
            level = []
            next = []
            for tmp in queue:
                level.append(tmp.val)
                if tmp.left:
                    next.append(tmp.left)
                if tmp.right:
                    next.append(tmp.right)
            if k & 1 == 0:
                res.append(level)
            else:
                level.reverse()
                res.append(level)
            queue = next[:]
            k = k + 1
        return res

2019.3.21