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

题目描述

给定一个二叉树的根节点 root ，返回其按层序遍历得到的节点值。
（即逐层地，从左到右访问所有节点）。

示例

输入:

    3
   / \
  9  20
    /  \
   15   7

输出:

[
  [3],
  [9,20],
  [15,7]
]

题解1：广度优先搜索（BFS）

import java.util.*;

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> result = new ArrayList<>();
        if (root == null) {
            return result;
        }
        
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        
        while (!queue.isEmpty()) {
            int size = queue.size();
            List<Integer> levelValues = new ArrayList<>();
            
            for (int i = 0; i < size; i++) {
                TreeNode node = queue.poll();
                levelValues.add(node.val);
                
                if (node.left != null) {
                    queue.offer(node.left);
                }
                if (node.right != null) {
                    queue.offer(node.right);
                }
            }
            
            result.add(levelValues);
        }
        
        return result;
    }
}

解决思路及复杂度分析：

• 这道题可以使用广度优先搜索（BFS）的思想解决。
• 首先定义一个队列，将二叉树的根节点加入队列。
• 在进行循环遍历时，首先获取当前队列的大小，表示当前层的节点个数。
• 然后从队列中取出当前层的所有节点，并将它们的值存入一个临时列表，同时将它们的左右子节点（如果存在）加入队列。
• 每层结束后，将临时列表添加到结果列表中。
• 最终返回结果列表即可。

该解法的时间复杂度为 O(N)，其中 N 是二叉树的节点数量，需要遍历所有节点。空间复杂度为 O(M)，其中 M 是二叉树的最大层级数，存储结果需要额外的空间，最坏情况下为 N/2，即完全二叉树的层级数。

题解2：深度优先搜索（DFS）

import java.util.*;

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

解决思路及复杂度分析：

• 另一种解法是使用深度优先搜索（DFS）的思想。
• 递归地进行深度遍历，同时将每个节点的值添加到对应层级的列表中。
• 遍历到当前节点时，根据层级信息判断是否需要在结果列表中添加新的层级列表，然后将节点的值添加到对应层级列表中。
• 最终返回结果列表即可。

该解法的时间复杂度为 O(N)，其中 N 是二叉树的节点数量，需要遍历所有节点。空间复杂度为 O(M)，其中 M 是二叉树的最大层级数，存储结果需要额外的空间，最坏情况下为 N/2，即完全二叉树的层级数。