Leetcode2028. 找出缺失的观测数据

Every day a Leetcode

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

解法1：模拟

统计当前 m 个元素的总和 curSum = sum(rolls)，总共 m+n 个元素和为 total = (m + n) * mean。

排除 2 种情况：

1. total - curSum > 6 * n：n 个骰子全丢 6 都补不了 total 和 curSum 的差距。
2. total - curSum < 1 * n：n 个骰子的最小点数和为 n，如果 total - curSum 小于 n，说明数据不合法。

我的答案数组的构造方法：

每个骰子先分配 (total - curSum) / n 个点数，再将 (total - curSum) % n 个骰子的点数加 1。

代码：

/** @lc app=leetcode.cn id=2028 lang=cpp** [2028] 找出缺失的观测数据*/// @lc code=start
class Solution
{
public:vector<int> missingRolls(vector<int> &rolls, int mean, int n){int m = rolls.size();int curSum = accumulate(rolls.begin(), rolls.end(), 0);int total = (m + n) * mean;if (total - curSum > 6 * n || total - curSum < 1 * n)return {};vector<int> ans(n, (total - curSum) / n);for (int i = 0; i < (total - curSum) % n; i++)ans[i]++;return ans;}
};
// @lc code=end

结果：

复杂度分析：

时间复杂度：O(m)，其中 m 是数字 rolls 的长度。

空间复杂度：O(n)。