Find Missing Observations Python Solution

lcpy gen -n 2028   # by problem number
lcpy gen -s find_missing_observations   # by problem name

Input: rolls = [3,2,4,3], mean = 4, n = 2
Output: [6,6]
Explanation: The mean of all n + m rolls is (3 + 2 + 4 + 3 + 6 + 6) / 6 = 4.

Input: rolls = [1,5,6], mean = 3, n = 4
Output: [2,3,2,2]
Explanation: The mean of all n + m rolls is (1 + 5 + 6 + 2 + 3 + 2 + 2) / 7 = 3.

Input: rolls = [1,2,3,4], mean = 6, n = 4
Output: []
Explanation: It is impossible for the mean to be 6 no matter what the 4 missing rolls are.

class Solution:
    # Time: O(m + n)
    # Space: O(n)
    def missing_rolls(self, rolls: list[int], mean: int, n: int) -> list[int]:
        target = mean * (len(rolls) + n) - sum(rolls)
        if target < n or target > 6 * n:
            return []
        base, extra = divmod(target, n)
        return [base + 1] * extra + [base] * (n - extra)