【二叉树打卡2】从上往下打印二叉树

原创

帅地玩编程 2020-11-26 17:26:50 ©著作权

文章标签 JAVA 文章分类 Java 后端开发

©著作权归作者所有：来自51CTO博客作者帅地玩编程的原创作品，请联系作者获取转载授权，否则将追究法律责任

【题目】

从上往下打印出二叉树的每个节点，同层节点从左至右打印。

【难度】

易

解答

这个像相当于二叉树四种遍历中的层序遍历了，其思想是采用广度优先遍历，借助一个辅助队列，步骤如下：

1、把二叉树的根节点 root 放进队列。

2、如果队列不为空，取出队列的一个节点，把这个节点的左右孩子放进队列中(为空的话就不用放)，然后打印这个节点。

3、一直重复步骤 2 ，直到队列为空，此时遍历结束，代码如下：

代码如下：


    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        ArrayList<Integer> list = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<>();
        if(root == null)
            return list;
        // 根放入队列
        queue.offer(root);
        while (!queue.isEmpty()) {
            TreeNode node =  queue.poll();
            list.add(node.val);
            if(node.left != null)
                queue.offer(node.left);
            if(node.right != null)
                queue.offer(node.right);
        }
        return list;
    }