题目分析:
运用DFS,通过调用递归函数helper实现代码运算。
定义一个levels为空集。
当前最高层数就是列表的长度 len(levels)。比较访问节点所在的层次 level 和当前最高层次 len(levels) 的大小,如果前者更大就向 levels 添加一个空列表。
将当前节点的内容插入到对应层的列表 levels[level] 中。
如果遇到子节点非空就更新当前节点,并进入下一层,不断进行递归调用。
代码如下:
class Solution(object):
def levelOrder(self, root):
"""
:type root: TreeNode
:rtype: List[List[int]]
"""
levels=[]
if not root:
return levels
def helper(node,level):
if len(levels)==level:
levels.append([])
levels[level].append(node.val)
if node.left:
helper(node.left,level+1)
if node.right:
helper(node.right,level+1)
helper(root,0)
return levels