力扣2028. 找出缺失的观测数据

题目：

现有一份 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

提交通过：