首页 > 其他 > 详细

LeetCode 102. 二叉树的层序遍历

时间:2020-11-18 21:32:33      阅读:40      评论:0      收藏:0      [点我收藏+]

102. 二叉树的层序遍历

Difficulty: 中等

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

示例:
二叉树:[3,9,20,null,null,15,7],

    3
   /   9  20
    /     15   7

返回其层次遍历结果:

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

Solution

Language: 全部题目

这是一道综合题,考察了栈、队列和树的知识,很经典。解法参考了???? Python | BFS Queue - 公瑾? (102) - LeetCode Discuss,那个size的用法很漂亮,放了多少到队列就拿多少出来。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
from collections import deque
class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        if not root:
            return []
        queue, res = deque([root]), []
?
        while queue:
            curLevel, size = [], len(queue)
            for i in range(size):
                node = queue.popleft()
                if node.left:
                    queue.append(node.left)
                if node.right:
                    queue.append(node.right)
                curLevel.append(node.val)
            res.append(curLevel)
        return res

LeetCode 102. 二叉树的层序遍历

原文:https://www.cnblogs.com/swordspoet/p/14001811.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!