Equilateral numbers

#include<iostream>
#include<vector>
#include <cmath>
#include <map>
using namespace std;

typedef long long ll;

#define lim 101010

ll dp[lim];
map<ll, bool> ek;

ll e(ll k){
    return k*(k+1)/2;
}
int main()
{
    ll m = 0;
    dp[0] = 0;
    for(ll i = 1; i< lim; ++i){
        ek[e(i)] = 1;
        dp[i] = 1e18;
        for (ll j =0; e(j) <= i; j++) {
            dp[i] = min(dp[i-e(j)] + 1, dp[i]);
        }
        m  = max(dp[i], m);
    }

    ll n;
    cin >> n;

    ll res = 4;
    for(ll i = 1; e(i) <= n; ++i){
        if(e(i) == n){
            cout << "1\n";
            return 0;
        }
        if(ek[n - e(i)]){
            cout << "2\n";
            return 0;            
        }
        if(n - e(i) < lim){
            res = min(res, dp[n-e(i)]+1);
        }
    }
    cout << res << endl;
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: WRONG ANSWER

Test 7

Verdict: ACCEPTED

Test 8

Verdict: WRONG ANSWER

Test 9

Verdict: WRONG ANSWER

Test 10

Verdict: ACCEPTED

Test 11

Verdict: WRONG ANSWER

Test 12

Verdict: WRONG ANSWER