二叉树层次遍历（算法）

public ArrayList<ArrayList<Integer>> LevelQuery(TreeNode root){
    //存放返回结果
    ArrayList<ArrayList<Integer>> restList = new ArrayList<>();
    
    if(root==null)
        return restList;
    
    //存放当前层节点
    ArrayList<TreeNode> level = new ArrayList<>();
    level.add(root);

    while(!level.isEmpty()){
        //存放当前层结果
        ArrayList<Integer> levelRest = new ArrayList<>();
        //存放下一层节点 
        ArrayList<TreeNode> nextNode = new ArrayList<>();
        for(TreeNode node:level){
            levelRest.add(node.val);
            if(node.left!=null)
                nextNode.add(node.left);
            if(node.right!=null)
                nextNode.add(node.right);
        }
        level = nextNode;//替换到处理下一层
        restList.add(levelRest);
    }
    return restList;
}

代码分析

进行每层遍历，将每层的结果存入List , 输出结果应为【【1】，【2，3】，【4，5】】