面试题32 - II. 从上到下打印二叉树 II

从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。

思路:

使用BFS的思路进行存储二叉树中的节点,在打印,其中使用了队列的性质
首先将第一个结点放入队列q中,如果节点不为空,很容易的知道第一层的输出就是这个节点。把这个节点存入到第一层的结果中,接着遍历当前的左右节点,若果不是空的话就把它加入到临时数组out中,然后更新q = out,这时的q就是下一层的节点,以此类推,直到q中没有节点,说明遍历完成。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        if not root:
            return []
        res = []
        q = [root]
        while q:
            out = []
            temp = []

            for i in q:
                temp.append(i.val)
                if i.left:
                    out.append(i.left)
                if i.right:
                    out.append(i.right)
            res.append(temp)
            q = out
        return res

来自:力扣leetcode

发布了43 篇原创文章 · 获赞 1 · 访问量 1895

猜你喜欢

转载自blog.csdn.net/weixin_43455338/article/details/104649145