在每个树行中找最大值

/*
 * 题目思路：
 * 使用Java Stream实现，先用广度优先搜索（BFS）
 * 收集每一层的节点，再计算最大值。
 */
import java.util.*;
import java.util.stream.*;
 
public class Solution {
    public List<Integer> largestValues(TreeNode root) {
        List<Integer> result = new ArrayList<>();
        if (root == null) return result;
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            List<TreeNode> level = new ArrayList<>();
            while (!queue.isEmpty()) level.add(queue.poll());
            result.add(level.stream().mapToInt(node -> node.val).max().orElse(Integer.MIN_VALUE));
            level.forEach(node -> {
                if (node.left != null) queue.offer(node.left);
                if (node.right != null) queue.offer(node.right);
            });
        }
        return result;
    }
}
 
class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) { val = x; }
}