【python-leetcode102-树的宽度遍历】二叉树的层次遍历

时间:2022-07-23
本文章向大家介绍【python-leetcode102-树的宽度遍历】二叉树的层次遍历,主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。

问题描述:

给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。

例如: 给定二叉树: [3,9,20,null,null,15,7],

3 / 9 20 / 15 7 返回其层次遍历结果:

[ [3], [9,20], [15,7] ]

代码:

# 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 []
        queue=[root]
        res=[]
        while queue:
            tmp=[] 
            for i in range(len(queue)):
                node=queue.pop(0)
                tmp.append(node.val)
                if node.left:
                    queue.append(node.left)
                if node.right:
                    queue.append(node.right)
            res.append(tmp)
        return res