Sword32-II——从上到下打印二叉树 II

最新推荐文章于 2021-08-11 09:47:09 发布

gezq0601

最新推荐文章于 2021-08-11 09:47:09 发布

阅读量124

点赞数

分类专栏： August算法二刷文章标签：队列二叉树 java leetcode 算法

本文链接：https://blog.csdn.net/A13777852949/article/details/119419823

版权

August算法二刷专栏收录该内容

75 篇文章 0 订阅

订阅专栏

Sword32-II——从上到下打印二叉树 II

方法1——BFS

思路：与上一题类似，只是输出结果格式发生了变化，只需要单独将每一层结果单独包装成一个List即可
特殊情况与临界分析：根节点为空
终止条件：队列中不存在元素
步骤：
- 定义一个存储结果res的ArrayList，便于校验中直接返回空ArrayList
- 根节点为空校验
- 定义一个辅助队列
- 定义一个移动节点
- 将移动节点（即根节点）入队
- 循环条件：队列非空
- while循环
  - 定义一个ArrayList用于存储每一层结果
  - for循环，因为queue的size存入元素后将会变化，因此需要记录当前queue的size大小，因此将i直接记录queue.size()，使得i依次减1即可，完成之后即弹出上一层的所有节点
    - 从队列中弹出一个元素
    - 将其加入到tmp的ArrayList中
    - 判断该节点左子节点是否非空，非空则入队
    - 判断该节点右子节点是否非空，非空则入队
  - 此时tmp的ArrayList已经装满了上一层的结果，将其加入到res中
- 返回结果（此处不需要转换为数组形式）

    public List<List<Integer>> levelOrder(TreeNode root) {
        // 定义结果List
        List<List<Integer>> res = new ArrayList<>();
        // 根节点为空校验
        if (root == null) {
            return res;
        }
        // 定义辅助队列
        Queue<TreeNode> queue = new LinkedList<>();
        // 定义移动节点
        TreeNode cur = root;
        // 将根节点入队
        queue.offer(cur);
        // while循环
        while (!queue.isEmpty()) {
            // 定义存储单层结果的ArrayList
            List<Integer> tmp = new ArrayList<>();
            // for循环
            for (int i = queue.size(); i > 0; i--) {
                // 弹出一个节点
                cur = queue.poll();
                // 将节点值加入到tmp中
                tmp.add(cur.val);
                // 判断其左子节点是否非空
                if (cur.left != null) {
                    // 非空则将左子节点入队
                    queue.offer(cur.left);
                }
                // 判断其右子节点是否非空
                if (cur.right != null) {
                    // 非空则将右子节点入队
                    queue.offer(cur.right);
                }
            }
            // for循环完毕，将tmp加入res中
            res.add(tmp);
        }
        // 返回结果
        return res;
    }