剑指 Offer 32 - II. 从上到下打印二叉树 II（Java、BFS）

最新推荐文章于 2024-07-15 22:27:18 发布

_Deser_

最新推荐文章于 2024-07-15 22:27:18 发布

阅读量96

点赞数 2

文章标签： java leetcode 数据结构算法 b树

本文链接：https://blog.csdn.net/weixin_45512600/article/details/129773852

版权

原题地址

剑指 Offer 32 - II. 从上到下打印二叉树 II

题目描述

在这里插入图片描述

题目思路

由上到下，自左向右遍历，即为二叉树的广度优先搜索（BFS），可以由队列（先进先出）来实现。
每层打印一行，即将本层全部节点打印到一行，并将下一层全部节点加入队列，以此类推，即可分为多行打印。

代码实现

广度优先搜索（BFS）

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        if(root == null) return new ArrayList(){};
        Queue<TreeNode> queue = new LinkedList<>();
        
        queue.add(root);
        List<List<Integer>> res = new ArrayList<>();
        while(!queue.isEmpty()){
            List<Integer> temp = new ArrayList<>();
            for(int i = queue.size(); i > 0; i--){
            	//由高位向低位遍历队列，防止队列长度变化影响
                TreeNode node = queue.poll();
                temp.add(node.val);

				//将下层节点入队
                if(node.left != null) queue.add(node.left);
                if(node.right != null) queue.add(node.right);             
            }
            res.add(temp);
        }

        return res;
    }
}