Equilateral numbers

#include <bits/stdc++.h>
 
// g++ -std=c++17 e.cpp -o e
using namespace std;
unordered_map<int64_t, int> d;
int hae(int64_t n){
    //cout << n << endl;
    if (d.count(n) > 0) return d[n];
    if (n==0) return 0;
    int64_t b = (-1+sqrt(1+8*n))/2;
    while((b*(b+1))/2 < n){
        b += 1;
    }
    if ((b*(b+1))/2 == n){
        return 1;
    }
    if((b*(b+1))/2 > n) {
        b--;
    }
    int mi = 1000000;
    int ll = sqrt(b);
    while(ll > 0){
        int h = hae(n-(b*(b+1))/2);
        mi = min(mi, h);
        --b;
        --ll;
    }
    d[n] = mi+1;
    return mi+1;
}
 
int main() {
    int64_t n; cin >> n;
    auto ans = hae(n);
    cout << ans << endl;
    return 0;
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: ACCEPTED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: TIME LIMIT EXCEEDED

Test 9

Verdict: TIME LIMIT EXCEEDED

Test 10

Verdict: ACCEPTED

Test 11

Verdict: TIME LIMIT EXCEEDED

Test 12

Verdict: TIME LIMIT EXCEEDED