Pussit

#!/usr/bin/env python3
 
def eka_tasajako(pusseja, palloja):
    for pussia in range(pusseja - 1, 1, -1):
        vajaapusseja = pusseja - pussia
        p = palloja
        while p > 0:
            if p % pussia == 0:
                return pussia
            p -= vajaapusseja
    return None
 
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
 
    n_pusseja = palloja % pusseja
    n_minus1_pusseja = pusseja - n_pusseja
    tasapusseja = eka_tasajako(pusseja, palloja)
    if tasapusseja is not None:
        turhia_nostoja_max = min(pusseja - tasapusseja, n_minus1_pusseja)
    else:
        turhia_nostoja_max = n_minus1_pusseja
 
    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: WRONG ANSWER

Test 2

Group: 2

Verdict: WRONG ANSWER

Test 3

Group: 3

Verdict: TIME LIMIT EXCEEDED