2028. 找出缺失的观测数据

from typing import List


class Solution:
    def missingRolls(self, rolls: List[int], mean: int, n: int) -> List[int]:
        ans = (len(rolls) + n) * mean - sum(rolls)

        if n <= ans < 6 * n:
            return (ans - ans // n * n) * [ans // n + 1] + ((ans // n + 1) * n - ans) * [ans // n]
        elif ans == 6 * n:
            return [6] * n
        else:
            return []