树与图中等2 种解法

#102二叉树的层序遍历

按从上到下、每层从左到右的顺序返回节点值。

#二叉树#广度优先搜索#深度优先搜索

解题主线

01

BFS 在每轮开始时固定当前队列长度,即可准确切分层。

02

DFS 可用深度作为结果列表下标。

解法 1队列层序遍历

每轮消费队列中已有的整层节点,并把下一层加入队尾。

时间复杂度

O(n)

空间复杂度

O(w)

102. 二叉树的层序遍历 · 队列层序遍历
import java.util.*;

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;

    TreeNode(int val) {
        this.val = val;
    }
}

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> result = new ArrayList<>();
        if (root == null) return result;
        Deque<TreeNode> queue = new ArrayDeque<>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            int size = queue.size();
            List<Integer> level = new ArrayList<>(size);
            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);
            }
            result.add(level);
        }
        return result;
    }
}

每轮消费队列中已有的整层节点,并把下一层加入队尾。

解法 2按深度递归

先序 DFS 时把节点加入与 depth 对应的列表。

时间复杂度

O(n)

空间复杂度

O(h),不计返回结果

102. 二叉树的层序遍历 · 按深度递归
import java.util.*;

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;

    TreeNode(int val) {
        this.val = val;
    }
}

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> result = new ArrayList<>();
        dfs(root, 0, result);
        return result;
    }

    private void dfs(TreeNode node, int depth, List<List<Integer>> result) {
        if (node == null) return;
        if (depth == result.size()) result.add(new ArrayList<>());
        result.get(depth).add(node.val);
        dfs(node.left, depth + 1, result);
        dfs(node.right, depth + 1, result);
    }
}

先序 DFS 时把节点加入与 depth 对应的列表。

边界与易错点

  • 遍历当前层时队列会动态增长,循环上界必须提前保存。
整理来源

由旧仓库源码复核、去重并整理;展示代码已按 Java 21 语义修正明显问题。

leetcode/src/main/java/tree/Q102_levelOrder.java