找出缺失的观测数据

/*
 题目思路：
 1. 计算 m 次投掷的和 currentSum。
 2. 计算 n + m 次投掷的总和 totalSum = (n + m) * mean。
 3. 计算缺失的 n 次投掷的和 missingSum = totalSum - currentSum。
 4. 检查 missingSum 是否在 n 和 6 * n 之间，如果不在，返回空数组。
 5. 使用 Java Stream 将 missingSum 均匀分配给 n 个数，每个数在 1 和 6 之间。
 */

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

public class MissingObservationsStream {
    public static int[] missingRolls(int[] rolls, int mean, int n) {
        int m = rolls.length;
        int currentSum = Arrays.stream(rolls).sum();
        int totalSum = (n + m) * mean;
        int missingSum = totalSum - currentSum;
        if (missingSum < n || missingSum > 6 * n) {
            return new int[0];
        }
        int average = missingSum / n;
        int extra = missingSum % n;
        return IntStream.range(0, n)
                .map(i -> average + (i < extra ? 1 : 0))
                .toArray();
    }
}