剑指--从上到下打印二叉树

最新推荐文章于 2023-02-20 09:38:30 发布

原创最新推荐文章于 2023-02-20 09:38:30 发布 · 139 阅读

0 ·

CC 4.0 BY-SA版权

剑指专栏收录该内容

77 篇文章

订阅专栏

本文详细介绍了如何使用队列实现二叉树的层序遍历，通过具体算法流程和代码示例，展示了从上到下打印二叉树的过程。适用于初学者理解和掌握二叉树的遍历技巧。

剑指–从上到下打印二叉树

这个题目和这个题目是很相似的，现在附上那个链接：剑指–从上到下打印二叉树 II
以上这个链接的题目呢，也是打印树，就是这个一层单独放到一个数组中，最后在统一放到一个结果数组中。。

1，题目：

在这里插入图片描述
2，思路：

层序遍历（用队列）：
（自己写的）

算法流程：

1.特例处理：当树的根节点为空，则直接返回空列表 [] ；
2.初始化：打印结果列表 res = [] ，包含根节点的队列 queue = [root] ；
3.BFS 循环：当队列 queue 为空时跳出；
4.出队：队首元素出队，记为 node；
5.打印：将 node.val 添加至列表 ans 尾部；
6.添加子节点：若 node 的左（右）子节点不为空，则将左（右）子节点加入队列 queue ；
7.返回值：返回打印结果列表 res 即可。

下面是对应的图解：

在这里插入图片描述

3，代码：

层序遍历（用队列）：

自己写的：

class Solution {
    public int[] levelOrder(TreeNode root) {
    /*
    算法流程：
1.特例处理： 当树的根节点为空，则直接返回空列表 [] ；
2.初始化： 打印结果列表 res = [] ，包含根节点的队列 queue = [root] ；
3.BFS 循环： 当队列 queue 为空时跳出；
4.出队： 队首元素出队，记为 node；
5.打印： 将 node.val 添加至列表 ans 尾部；
6.添加子节点： 若 node 的左（右）子节点不为空，则将左（右）子节点加入队列 queue ；
7.返回值： 返回打印结果列表 res 即可。


    */
        Queue<TreeNode> queue = new LinkedList<>();
        ArrayList<Integer> ans = new ArrayList<>();
        if(root != null)
            queue.add(root);
        while(!queue.isEmpty()){
            for(int i = queue.size(); i>0; i--){//因为队列是先进先出。所以从末尾开始弹出
                TreeNode node = queue.poll();
                ans.add(node.val);
                if(node.left != null)
                    queue.add(node.left);
                if(node.right != null)
                    queue.add(node.right);

            }
        }
        int[] res = new int[ans.size()];
        for(int i = 0; i < ans.size(); i++)
            res[i] = ans.get(i);


        return res;

    }
}