把二叉树打印成多行

最新推荐文章于 2020-09-10 00:01:09 发布

qq_43592587

最新推荐文章于 2020-09-10 00:01:09 发布

阅读量88

点赞数

分类专栏：剑指offer 文章标签：算法 python

本文链接：https://blog.csdn.net/qq_43592587/article/details/104669182

版权

剑指offer 专栏收录该内容

36 篇文章 0 订阅

订阅专栏

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

思路一：
BFS

代码：

class Solution:
    # 返回二维列表[[1,2],[4,5]]
    def Print(self, pRoot):
        queue, res = [], []
        queue.append(pRoot)
        while queue:
            lst = []
            index = 0
            length = len(queue)
            while index < length:
                p = queue.pop(0)
                if p:
                    queue.append(p.left)
                    queue.append(p.right)
                    lst.append(p.val)
                index += 1
            if lst: res.append(lst)
        return res

思路二：
递归

代码：

class Solution:
    # 返回二维列表[[1,2],[4,5]]
    def Print(self, pRoot):
        res = []
        depth = 1
        self.helper(pRoot, depth, res)
        return res
    
    def helper(self, p, depth, res):
        if not p: return None
        if depth > len(res):
            res.append([])
        res[depth-1].append(p.val)
        self.helper(p.left, depth+1, res)
        self.helper(p.right, depth+1, res)