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

[LeetCode]题解(python):102- Binary Tree Level Order Traversal

时间:2016-03-07 18:45:17      阅读:188      评论:0      收藏:0      [点我收藏+]

标签:

题目来源:

  https://leetcode.com/problems/binary-tree-level-order-traversal/


 

题意分析:

  宽度优先搜索一颗二叉树,其中同一层的放到同一个list里面。比如:

  3
   /   9  20
    /     15   7
返回
[
  [3],
  [9,20],
  [15,7]
]

 

题目思路:

  新定义一个函数,加多一个层数参数,如果目前答案的列表答案个数等于层数,那么当前层数的列表append新元素,否者加新的层数。


 

代码(python):

  

技术分享
# 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]]
        """
        ans = []
        def bfs(root,level):
            if root != None:
                if len(ans) < level + 1:
                    ans.append([])
                ans[level].append(root.val)
                bfs(root.left,level+1)
                bfs(root.right,level+1)
        bfs(root,0)
        return ans
        
View Code

 

[LeetCode]题解(python):102- Binary Tree Level Order Traversal

标签:

原文地址:http://www.cnblogs.com/chruny/p/5251414.html

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