[LeetCode]题解（python）：102- Binary Tree Level Order Traversal

Posted 2020-06-18 Ry_Chen

题目来源：

　　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