题目来源:
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