心路历程:一开始想的是使用bfs但是有很多的细节没考虑到(不知道如何去处理一层只有一个节点的情况),然后又想到了使用递归回溯的来解决。
解题思路:使用一个总的List来记录所有的List集合,然后从根节点开始调用函数(根节点所在的层数是总list的下表0),在处理一个节点的时候先看该层的list是否被建立过,如果没有就建立一个list放入。然后向这个list中添加当前的节点,把该节点的两个子节点和他们所在的层数作为参数进行函数调用。当前节点为空时直接返回程序。
代码:
List<List<Integer>> listSum=new ArrayList<>();
public List<List<Integer>> levelOrder(TreeNode root) {
next(root,0);
return listSum;
}
public void next(TreeNode root,int level){
if(root==null){
return ;
}
List<Integer> list;
if(listSum.size()>level){
list=listSum.get(level);
}else{
list=new ArrayList<>();
listSum.add(list);
}
list.add(root.val);
next(root.left,level+1);
next(root.right,level+1);
listSum.set(level,list);
}