結果

問題 No.2510 Six Cube Sum Counting
コンテスト
ユーザー umimel
提出日時 2023-10-21 10:11:45
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
MLE  
実行時間 -
コード長 1,402 bytes
コンパイル時間 1,864 ms
コンパイル使用メモリ 178,328 KB
実行使用メモリ 518,748 KB
最終ジャッジ日時 2024-09-21 11:18:03
合計ジャッジ時間 7,801 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 MLE * 1 -- * 1
other -- * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using ll = long long;
using pll = pair<ll, ll>;
#define drep(i, cc, n) for (ll i = (cc); i <= (n); ++i)
#define rep(i, n) drep(i, 0, n - 1)
#define all(a) (a).begin(), (a).end()
#define pb push_back
#define fi first
#define se second
mt19937_64 rng(chrono::system_clock::now().time_since_epoch().count());
const ll MOD1000000007 = 1000000007;
const ll MOD998244353 = 998244353;
const ll MOD[3] = {999727999, 1070777777, 1000000007};
const ll LINF = 1LL << 60LL;
const int IINF = (1 << 30) - 1;


void solve(){
    int x; cin >> x;
    unordered_map<int, vector<short>> mp1, mp2;
    int MAX = 300*300*300*3;
    for(int a=0; a<=300; a++) for(int b=a; b<=300; b++) for(int c=b; c<=300; c++){
        int abc = a*a*a+b*b*b+c*c*c;
        if(abc+MAX < x) continue;
        
        if(abc<=x/2){
            mp1[abc].pb((short)c);
        }
        if(x/2<=x && abc<=x){
            mp2[abc].pb((short)a);
        }
    }
    ll ans = 0;
    for(auto itr=mp1.begin(); itr!=mp1.end(); itr++){
        int y = x-(*itr).fi;
        if(mp2.find(y)==mp2.end()) continue;
        for(short c : (*itr).se){
            ans += (ll)mp2[y].size() - (ll)(lower_bound(all(mp2[y]), c)-mp2[y].begin());
        }
    }

    cout << ans << endl;
}

int main(){
    cin.tie(nullptr);
    ios::sync_with_stdio(false);
    
    int T=1;
    //cin >> T;
    while(T--) solve();
}
0