LeetCode102：二叉树的层序遍历（队列与递归）

二叉树的层序遍历

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

广度优先遍历BFS：

    /**
     * 二叉树的层序遍历：队列（先出先进，每次入队一层，出队一层的节点）：广度优先遍历
     * 时间复杂度：O(n)
     * 空间复杂度:O(n)
     * @param root
     * @return
     */
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();
        if(root == null){
            return res;
        }
        Queue<TreeNode> queue = new LinkedList<>();
        //根节点入队列
        queue.add(root);
        while (queue.size() >0){
            //队列中还有元素
            //获取队列元素的个数，队列元素中的个数相当于该层的节点数量
            int number = queue.size();
            List<Integer> temp = new ArrayList<>();
            //将队列中节点(该层的所以有节点)出队，放入temp集合中，
            // 如果节点的左右子树不为空，并将孩子节点（下层的节点）入队
            for(int i = 0 ;i<number;i++){
                //出队,将节点从队列弹出
                TreeNode tree = queue.poll();
                temp.add(tree.val);
                if(tree.left != null){
                    queue.add(tree.left);
                }
                if(tree.right != null){
                    queue.add(tree.right);
                }
            }
            //将temp加入最终返回结果中
            res.add(temp);
        }
        return res;
    }

深度优先遍历DFS：

 /**
     * 深度优先遍历dfs： DFS 不是按照层次遍历的,
     * 为了让递归的过程中同一层的节点放到同一个列表中，在递归时要记录每个节点的深度 level。
     * 递归到新节点要把该节点放入 level 对应列表的末尾。
     * @param res
     * @param node
     * @param level
     * @return
     */
    private void dfs(List<List<Integer>> res,TreeNode node,int level){
        //当遍历到一个新的深度 level，而最终结果 res 中还没有创建 level 对应的列表时，
        // 应该在 res 中新建一个列表用来保存该 level 的所有节点。
        if(res.size()-1<level){
            res.add(new ArrayList<>());
        }
        //把该节点放入level对应列表的末尾
        res.get(level).add(node.val);
        if(node.left != null){
            dfs(res,node.left,level+1);
        }
        if(node.right != null){
            dfs(res,node.right,level+1);
        }
    }
    /**
     * 二叉树的层次遍历：深度优先遍历DFS：递归
     * @param root
     * @return
     */
    public List<List<Integer>> levelOrder2(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();
        if(root != null){
            dfs(res,root,0);
        }
        return res;
    }