题目

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

示例 1：

输入：root = [3,9,20,null,null,15,7]
输出：[[3],[9,20],[15,7]]

示例 2：

输入：root = [1]
输出：[[1]]

示例 3：

输入：root = []
输出：[]

提示：

• 树中节点数目在范围 [0, 2000] 内
• -1000 <= Node.val <= 1000

题解

public List<List<Integer>> levelOrder(TreeNode root) {
    // 按照层级缓存数字
    Map<Integer, List<Integer>> cache = new HashMap<>(8);

    BiConsumer<TreeNode, Integer> recursion = new BiConsumer<TreeNode, Integer>() {
        @Override
        public void accept(TreeNode node, Integer layer) {
            if (null == node) {
                return;
            }
            // 将数字加入对应层级
            cache.merge(layer, new ArrayList<>(Collections.singletonList(node.val)), (o, n) -> {
                o.add(node.val);

                return o;
            });
            // 遍历左树
            this.accept(node.left, layer + 1);
            // 遍歷右樹
            this.accept(node.right, layer + 1);
        }
    };
    recursion.accept(root, 0);

    return cache.keySet().stream().sorted().map(cache::get).collect(Collectors.toList());
}