LeetCode 2028. 找出缺失的观测数据题解

2028. 找出缺失的观测数据题解

题目来源：2028. 找出缺失的观测数据

2022.03.27 每日一题

LeetCode 题解持续更新中Github仓库地址 CSDN博客地址

今天的题目就是一个简单的构造题

根据题目描述，数组 rolls 的长度为 m，记录了 m 个观测数据，还有 n 个观测数据缺失，共有 n+m 个观测数据。由于所有观测数据的平均值为mean，因此所有观测数据之和为 mean×(n+m)。

根据所有观测数据之和与数组 rolls 中的 m 个观测数据，可知缺失的 n 个观测数据之和。将缺失的 n 个观测数据之和记为 ans

由于骰子的点数只有 1、2、3、4、5、6 六个值，因此 将 ans 平均下来的每个值都应该在这个范围之中

因此有$ n<=ans<=6*n$，如果ans不在这个范围之中，说明无法分配点数，返回空数组

若在这个范围之中，苦于先将点数均分，再将剩余点数随机分配给结果数组，即可得到最终答案

class Solution {
public:
    vector<int> missingRolls(vector<int> &rolls, int mean, int n) {
        // 统计数组 roll 中的元素和
        int cnt = 0;
        // 获取 m 的个数
        int m = rolls.size();
        // 求得总和
        for (int r: rolls) cnt += r;
        // 由 roll 数组的总和求出 结果数组的总和
        int ans = mean * (m + n) - cnt;
        // 由于骰子的点数在 1 - 6 之间，若是不满足范围即可直接返回空数组
        if (ans < n || ans > 6 * n) return {};
        // 将剩余的数量均分
        vector<int> res(n, ans / n);
        // 再统计均分以后还剩余多少
        cnt = ans % n;
        // 将剩余的数量进行均分
        for (int i = 0; cnt != 0; i++) {
            res[i]++;
            cnt--;
        }
        return res;
    }
};

class Solution {
    public int[] missingRolls(int[] rolls, int mean, int n) {
        // 统计数组 roll 中的元素和
        int cnt = 0;
        // 获取 m 的个数
        int m = rolls.length;
        // 求得总和
        for (int r : rolls) cnt += r;
        // 由 roll 数组的总和求出 结果数组的总和
        int ans = mean * (m + n) - cnt;
        // 由于骰子的点数在 1 - 6 之间，若是不满足范围即可直接返回空数组
        if (ans < n || ans > 6 * n) return new int[0];
        // 将剩余的数量均分
        int[] res = new int[n];
        Arrays.fill(res, ans / n);
        // 再统计均分以后还剩余多少
        cnt = ans % n;
        // 将剩余的数量进行均分
        for (int i = 0; cnt != 0; i++) {
            res[i]++;
            cnt--;
        }
        return res;
    }
}