每日一题-找出缺失的观测数据

发表于 分类于 Waline：

2028. 找出缺失的观测数据

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/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 整除。

2-Dice-Icon

思路

由于题目仅需返回满足答案的数据，直接模拟就好。

  1. 求出缺失数据的点数总和sum
  2. $avg=sum/n$为这些点数的平均值，$avg*n-sum$为还需补充的点数，遍历答案数据补1即可。
  • 当sum < n || sum > 6*n时没和符合要求的答案，骰子不可能掷出大于6或负数。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
class Solution {
	public:
	    vector<int> missingRolls(vector<int>& rolls, int mean, int n) {
		int m = rolls.size();
		int sum = mean*(m+n)-accumulate(rolls.begin(),rolls.end(),0);
		if (sum < n || sum > 6*n) {
			return {
			}
			;
		}
		int avg = sum/n;
		vector<int> ans (n,avg);
		int diff = sum-avg*n;
		int i = 0 ;
		while(diff) {
			++ans[i];
			--diff;
			++i;
		}
		return ans;
	}
}
;