题目:
现有一份 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
整除。
思考:
先计算出缺失的n个数据之和s,然后判断是否存在符合条件的数组(每个元素的取值范围为1至6),若存在则计算数组,否则返回空数组。
具体计算数组的过程如下:
- 数组初始化为n个1,记s为剩余可分配的值,s = s(原) - n
- 循环(当s大于0):给当前数组元素分配1,s减一,移到下一元素
- 最后返回数组
代码如下:
class Solution:
def missingRolls(self, rolls: List[int], mean: int, n: int) -> List[int]:
s = (len(rolls) + n) * mean - sum(rolls) # 缺失数据的和
if s > 6 * n or s < 1 * n:
return []
ans = [1 for _ in range(0, n)]
s -= n
i = 0
while s > 0:
ans[i] += 1
s -= 1
if i == n-1:
i = 0
else:
i += 1
return ans
提交通过: