leetcode笔记----102.二叉树的层序遍历

题目描述：给你一个二叉树，请你返回其按 层序遍历 得到的节点值。 （即逐层地，从左到右访问所有节点）。

一般的二叉树层次遍历很简单，用一个队列存储节点，每次出队时将其子节点再入队即可，但本题需要返回一个List<List< Integer >>，也就是说要按照层次分层存储节点值。经过在纸上的比比划划，我得到的思路是：首先根节点入列表，然后维护两个列表，一个存储当前层，一个存储下一层，当当前层遍历完后，把下一层的节点导入当前层继续遍历。代码如下

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {

         List<TreeNode> allnode = new ArrayList();
         List<List<Integer>> res = new ArrayList();
         allnode.add(root);
         List<Integer> tempval = new ArrayList();
         List<TreeNode> temp = new ArrayList();
         if(root == null){return res;}
         while(allnode.size()!=0 || temp.size()!=0){
             if(allnode.size()!=0){                
                TreeNode p =allnode.get(0);
                if(p.left!=null) {temp.add(p.left);}
                if(p.right!=null) {temp.add(p.right);}
                tempval.add(p.val);
                allnode.remove(0);
             }
             else{                                  //当前层节点都加入完毕，把存在临时列表的下一层节点倒入
             List<Integer> finaltemp = new ArrayList<>();
                    for (int i = 0; i < tempval.size(); i++) {
                        finaltemp.add(tempval.get(i));
                    }
                res.add(finaltemp);
                tempval.clear();
                while(temp.size()!=0){
                    TreeNode p1 = temp.get(0);
                    allnode.add(p1);
                    temp.remove(0);
                }
             }
         }
        List<Integer> finaltemp = new ArrayList<>();
        for (int i = 0; i < tempval.size(); i++) {
            finaltemp.add(tempval.get(i));
        }
        res.add(finaltemp);
         return res;
    }
}

由于只是初步的想法，所以有很多不成熟的问题，浪费了很多空间。仔细想一想其实只需一个列表就能完成所有操作：在循环一开始时记录当前队列的size，则这个size就是当前层所含有的所有节点数目，只需要把这些遍历即可，至于加入子节点也完全可以同时进行，这样就节省了一大批空间和时间。