题目描述

题目简化

这题目说的又臭又长，我简化一下题目，就是玩骰子，每次记录下来骰子显示的数。一共玩X次，但其中n次的数据丢了，仅知道 X次所显示数的平均值mean，让你求出来其中丢失n次的各值是多少，多种组合返回其中一种即可，顺序也可以无视。

思路分析

初始方法

题目给出了总的平均值，我们只需要知道玩了几次骰子，然后乘以平均值就知道总共的数值和，从而知道缺失的那部分的数值和，然后只需要从1开始逐个加1 ，直到这部分的和加到等于缺失的总数值时即可。

拿示例1 来说:

rolls = [1,5,6], mean = 3, n = 4

那么一共就是玩了 3+4 = 7次骰子。
总共的数值就是 7* 3 = 21.
缺失部分的总数值就是 21- （1 + 5 +6） = 9
缺失的次数是 4次，4次的总数值是9，所以建立数组 res = [1,1,1,1]
开始依次加1，直到数组和等于9.。
最后 res = [3,2,2,2]

最后记得处理无解的情况，假如缺失的数值总和为15，而题目所给的缺失次数n = 2，此时就算缺失的那两次都显示的6，也只有12个数，无法达到缺失的15，此时无解。相反，如果缺失的数值为2，而缺失的次数为5，如果5次都显示1，也大于缺失的数值2，此时也是无解。

按照上面的思路就可以了。

优化方法

其实在上面的方法中，res的设置不需要全部从1开始。

依旧是上面的那个例子，res的初始值设置可以为缺失的数值总和 / 缺失的次数。

所以上面的例子所设置的初始值就是 9 / 4 =2 向下取整。

从 res = [2,2,2,2] 开始逐个加1 直到符合条件。

完整代码

class Solution:
    def missingRolls(self, rolls: List[int], mean: int, n: int) -> List[int]:

        rool_sum = len(rolls) + n
        # rool_sum 总共投了多少次骰子

        cor_sum = sum(rolls)
        # cor_sum  现有数值总和

        num_sum = mean * rool_sum
        # num_sum 总共的数值和

        num_miss = num_sum - cor_sum
        # num_miss 缺失的数总和

        start_num =  num_miss / n
        # 分配到每一个数的初始值
        

        # 处理无解的情况
        if num_miss > n * 6 or num_miss < n * 1:
            return []
        res = [int(start_num)]*n

        res_sum = sum(res)

        temp = num_sum - cor_sum - res_sum
        # temp 还差多少个数 平均加到每一个上面就行了
        k = 0
        length = len(res)

        for z in range(temp):
            res[z] +=1
        return res