Pussit

#!/usr/bin/env python3
 
def eka_tasajako(pusseja, palloja):
    for pussia in range(pusseja, 1, -1):
        if palloja % pussia == 0:
            return pussia
 
def laske(pusseja, palloja, halutaan):
    if palloja < pusseja:
        tyhjia_pusseja = pusseja - palloja
        return tyhjia_pusseja + halutaan
    elif palloja == pusseja:
        return halutaan
 
    # palloja > pusseja
 
    if palloja % pusseja == 0:
        # voidaan jakaa tasan, ei turhia nostoja
        return halutaan
 
    if halutaan <= pusseja:
        # nostetaan aina eri pussista, ei huteja
        return halutaan
 
    min_palloja_per_pussi = palloja // pusseja
    helppoja_nostoja = pusseja * min_palloja_per_pussi
    if halutaan <= helppoja_nostoja:
        return halutaan
 
    tasapusseja = eka_tasajako(pusseja, palloja)
    turhia_nostoja_max = pusseja - tasapusseja
    return turhia_nostoja_max + halutaan
 
def main():
    for i in range(int(input())):
        n, m, k = map(int, input().strip().split())
        print(laske(n, m, k))
 
if __name__ == '__main__':
    main()

Test 1

Group: 1

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 40, in <module>
    main()
  File "input/code.py", line 37, in main
    print(laske(n, m, k))
  File "input/code.py", line 31, in laske
    turhia_nostoja_max = pusseja - tasapusseja
TypeError: unsupported operand type(s) for -: 'int' and 'NoneType'

Test 2

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 40, in <module>
    main()
  File "input/code.py", line 37, in main
    print(laske(n, m, k))
  File "input/code.py", line 31, in laske
    turhia_nostoja_max = pusseja - tasapusseja
TypeError: unsupported operand type(s) for -: 'int' and 'NoneType'

Test 3

Group: 3

Verdict: TIME LIMIT EXCEEDED