剑指offer—把二叉树打印成多行(Python版)

题目描述:

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

解题思路:

本题同“按之字形顺序打印二叉树”,使用队列存储数据。

代码实现:
# class TreeNode:
#    def __init__(self,x):
#        self.val = x
#        self.left = None
#        self.right = None
class Solution:
    def Print(self,pRoot):
        if pRoot == None:
            return []

        queue1 = [pRoot]
        queue2 = []
        ret = []
        while queue1 or queue2:
            if queue1:
                tmpRet = []
                while queue1:
                    tmpNode = queue1[0]
                    tmpRet.append(tmpNode.val)
                    del queue1[0]
                    if tmpNode.left:
                        queue2.append(tmpNode.left)
                    if tmpNode.right:
                        queue2.append(tmpNode.right)
                ret.append(tmpRet)

            if queue2:
                tmpRet = []
                while queue2:
                    tmpNode = queue2[0]
                    tmpRet.append(tmpNode.val)
                    del queue2[0]
                    if tmpNode.left:
                        queue1.append(tmpNode.left)
                    if tmpNode.right:
                        queue1.append(tmpNode.right)
                ret.append(tmpRet)
        return ret

你可能感兴趣的:(剑指offer)