彩灯装饰记录 I

// 题目思路：
// 1. 通过流的方式处理二叉树的层序遍历。
// 2. 使用一个队列来辅助遍历，将根节点放入队列。
// 3. 通过Stream的方式处理每一层的节点值，最后收集到一个列表中。

import java.util.*;
import java.util.stream.Collectors;

public class Solution {
    public List<Integer> levelOrder(TreeNode root) {
        if (root == null) return Collections.emptyList();
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        return queue.stream()
            .flatMap(node -> {
                List<TreeNode> children = new ArrayList<>();
                if (node.left != null) children.add(node.left);
                if (node.right != null) children.add(node.right);
                queue.addAll(children);
                return Stream.concat(Stream.of(node), children.stream());
            })
            .map(node -> node.val)
            .collect(Collectors.toList());
    }

    // Definition for a binary tree node.
    public static class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;
        TreeNode(int val) { this.val = val; }
    }
}