Equilateral numbers

from math import floor, sqrt


def e(n):
    if n <= 1:
        return n
    else:
        k = floor((sqrt(8 * n + 1) - 1) / 2 + 0.0001)
        while (k * (k + 1) / 2) > n:
            k -= 1
        return 1 + e(n - (k * (k + 1) / 2))

n = int(input())

print(int(e(n)))

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