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

    tasapusseja = eka_tasajako(pusseja, palloja)
    if tasapusseja is not None:
        turhia_nostoja_max = pusseja - tasapusseja
    else:
        n_pusseja = palloja % pusseja
        n_minus1_pusseja = pusseja - n_pusseja
        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