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

题目描述

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

image-20250420104928839

image-20250420104945150

image-20241107210303414

思路分析

解法一:层序遍历(推荐)

核心思路

  • 使用队列进行 BFS,按层收集节点。
  • 每一层遍历结束后加入结果列表。
  • 维持队列即可完成层序输出。


复杂度分析

  • 时间复杂度:O(n),其中 n 表示节点数。
  • 空间复杂度:O(n),队列在最坏情况下存储一层节点。
import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.List;
import java.util.Queue;

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();
        if (root == null) {
            return res;
        }

        Queue<TreeNode> queue = new ArrayDeque<>();
        queue.offer(root);

        while (!queue.isEmpty()) {
            int size = queue.size();
            List<Integer> level = new ArrayList<>();

            for (int i = 0; i < size; i++) {
                TreeNode node = queue.poll();
                level.add(node.val);

                if (node.left != null) {
                    queue.offer(node.left);
                }
                if (node.right != null) {
                    queue.offer(node.right);
                }
            }

            res.add(level);
        }

        return res;
    }
}
func levelOrder(root *TreeNode) [][]int {
	res := make([][]int, 0)
	if root == nil {
		return res
	}

	queue := []*TreeNode{root}
	for len(queue) > 0 {
		size := len(queue)
		level := make([]int, 0, size)
		for i := 0; i < size; i++ {
			node := queue[0]
			queue = queue[1:]
			level = append(level, node.Val)

			if node.Left != nil {
				queue = append(queue, node.Left)
			}
			if node.Right != nil {
				queue = append(queue, node.Right)
			}
		}
		res = append(res, level)
	}

	return res
}

相似题目

题目 难度 考察点
102. 二叉树的层序遍历 中等 BFS
103. 二叉树的锯齿形层序遍历 中等 BFS
107. 二叉树的层序遍历 II 中等 BFS
199. 二叉树的右视图 中等 BFS
515. 在每个树行中找最大值 中等 BFS
本文作者:
本文链接: https://hgnulb.github.io/blog/2023/32517451
版权声明: 本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议,转载请注明出处!