Equilateral numbers

import math

k = int(input())

def eq(n):
    return math.comb(n,2)

def inner(target):
    if target == 0:
        return 0

    i = 0
    e = eq(i)

    while e < target:
        i += 1
        e = eq(i)
    
    if e == target:
        return 1
    elif e > target:
        one_smaller = eq(i-1)
        diff = target - one_smaller
        return 1 + inner(diff)

    elif e < target:
        return 1 + inner(target-e)

print(inner(k))

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: WRONG ANSWER

Test 6

Verdict: WRONG ANSWER

Test 7

Verdict: WRONG ANSWER

Test 8

Verdict: WRONG ANSWER

Test 9

Verdict: WRONG ANSWER

Test 10

Verdict: ACCEPTED

Test 11

Verdict: WRONG ANSWER

Test 12

Verdict: WRONG ANSWER