Equilateral numbers

import math

k = int(input())


def eq(n):
    return (n*(n+1))/2



def inner(target):
    i = target
    e = eq(i)
    while e > target:
        e =eq(i)
        i-=1
    
    if e == target:
        return 1
    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: TIME LIMIT EXCEEDED

Test 7

Verdict: WRONG ANSWER

Test 8

Verdict: TIME LIMIT EXCEEDED

Test 9

Verdict: TIME LIMIT EXCEEDED

Test 10

Verdict: TIME LIMIT EXCEEDED

Test 11

Verdict: TIME LIMIT EXCEEDED

Test 12

Verdict: TIME LIMIT EXCEEDED