leetcode--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 整除。¹

• 示例：

# 示例 1
输入：rolls = [3,2,4,3], mean = 4, n = 2
输出：[6,6]
解释：所有 n + m 次投掷的平均值是 (3 + 2 + 4 + 3 + 6 + 6) / 6 = 4 。
# 示例 2
输入：rolls = [1,5,6], mean = 3, n = 4
输出：[2,3,2,2]
解释：所有 n + m 次投掷的平均值是 (1 + 5 + 6 + 2 + 3 + 2 + 2) / 7 = 3 。
# 示例 3
输入：rolls = [1,2,3,4], mean = 6, n = 4
输出：[]
解释：无论丢失的 4 次数据是什么，平均值都不可能是 6 。
# 示例 4
输入：rolls = [1], mean = 3, n = 1
输出：[5]
解释：所有 n + m 次投掷的平均值是 (1 + 5) / 2 = 3 。

• 提示：

1. m == rolls.length
2. 1 <= n, m <= 105
3. 1 <= rolls[i], mean <= 6

• 思路：

冗长版和简短版思路相同。首先先判断缺失的观测数据是不是在1-6之间，若不在其中，则直接返回空数组；若在1-6之间，则按照从前往后的顺序依次增加1。最后输出目标数组。

• 注意：该代码放在leetcode python中有的地方返回return不出来，可能是bug，调试的时候没问题，放上去就运行不出来，结果不一样。所以还是要用python3，python3不会遇到这种问题！！！！

• 解法一----冗长版：

class Solution:
    def missingRolls(self, rolls, mean, n):
        res = []
        sum_len = (n + len(rolls)) * mean
        left_sum = sum_len - sum(rolls)
        # 不符合要求在1-6之间
        leleft_sum = left_sum / n
        if leleft_sum > 6 or leleft_sum < 1:
            return res
        # 假设符合条件要求在1-6之间
        for i in range(n):
            res.append(int(leleft_sum))
        to = left_sum % n
        index = 0
        if to <= n:
            for i in range(to):
                res[index] += 1
                to -= 1
                index += 1
        else:
            less = to / n
            for i in range(n):
                res[index] += less
                to -= less
            if to <= n:
                for i in range(to):
                    res[index] += 1
                    to -= 1
        return res

• 解法二----简短版：

class Solution:
    def missingRolls(self, rolls, mean, n):
        m = len(rolls)
        nSum = (m + n) * mean - sum(rolls)
        if nSum > 6 * n or nSum < n:
            return []
        rest = nSum % n
        nMean = int(nSum / n)
        return [nMean for _ in range(n - rest)] + [nMean + 1 for _ in range(rest)]

• 总结：

1. 思路要清晰，不能丢小细节，比如之前把0改成1，条件就判断错误了，自然会报错；
2. 要注意写代码的格式，一定要规范整齐；
3. 列表表达式yyds。

---

1. 来源：力扣（LeetCode）
   链接：https://leetcode-cn.com/problems/find-missing-observations ↩︎