Bootstrap

力扣每日刷题

102. 二叉树的层序遍历

心路历程:一开始想的是使用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);
    }

;