2028. Find Missing Observations #

题目 #

现有一份 n + m 次投掷单个六面骰子的观测数据，骰子的每个面从 1 到 6 编号。观测数据中缺失了 n 份，你手上只拿到剩余 m 次投掷的数据。幸好你有之前计算过的这 n + m 次投掷数据的 平均值 。
给你一个长度为 m 的整数数组 rolls ，其中 rolls[i] 是第 i 次观测的值。同时给你两个整数 mean 和 n 。
返回一个长度为 n 的数组，包含所有缺失的观测数据，且满足这 n + m 次投掷的 平均值 是 mean 。如果存在多组符合要求的答案，只需要返回其中任意一组即可。如果不存在答案，返回一个空数组。
k 个数字的 平均值 为这些数字求和后再除以 k 。
注意 mean 是一个整数，所以 n + m 次投掷的总和需要被 n + m 整除。

思路 #

模拟 #

代码 #

模拟 #

class Solution {
    public int[] missingRolls(int[] rolls, int mean, int n) {
        int mSum = 0;
        for (int roll: rolls) mSum += roll;

        int nSum = mean * (n + rolls.length) - mSum;
        if (nSum < n || nSum > n * 6) return new int[0];

        int[] ans = new int[n];

        nSum -= n; // 保证每枚骰子的点数至少为1
        for (int i = 0; i < n; i++) {
            int given = nSum == 0 ? 0 : nSum > 5 ? 5 : nSum; // 贪心
            ans[i] = 1 + given;
            nSum -= given;
        }

        return nSum == 0 ? ans : new int[0];
    }
}

致谢 #

宫水三叶