码迷,mamicode.com
首页 > 编程语言 > 详细

LeetCode in Python 102. Binary Tree Level Order Traversal

时间:2019-07-11 20:48:15      阅读:127      评论:0      收藏:0      [点我收藏+]

标签:log   bsp   pytho   root   tps   return   pen   队列实现   its   

Given a binary tree, return the level order traversal of its nodes‘ values. (ie, from left to right, level by level).

For example:
Given binary tree [3,9,20,null,null,15,7],

    3
   /   9  20
    /     15   7

 

return its level order traversal as:

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

Solution:

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

class Solution(object):
    def levelOrder(self, root):
        """
        :type root: TreeNode
        :rtype: List[List[int]]
        """
        q, res = root and [root], []
        while q:
            size, level_path = len(q), []
            for i in range(size):
                node = q.pop(0)
                level_path.append(node.val)
                if node.left:
                    q.append(node.left)
                if node.right:
                    q.append(node.right)
            res.append(level_path)
        return res
        

最经典的队列实现BFS,注意size是作为for循环次数的依据,Python的 for node in q 的循环不适用,因为q在动态变化。

N-ary的树可参考559(https://www.cnblogs.com/lowkeysingsing/p/11152751.html

LeetCode in Python 102. Binary Tree Level Order Traversal

标签:log   bsp   pytho   root   tps   return   pen   队列实现   its   

原文地址:https://www.cnblogs.com/lowkeysingsing/p/11172103.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!