結果

問題 No.2510 Six Cube Sum Counting
コンテスト
ユーザー otoshigo
提出日時 2023-10-20 22:52:56
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
MLE  
実行時間 -
コード長 1,601 bytes
コンパイル時間 1,427 ms
コンパイル使用メモリ 110,520 KB
最終ジャッジ日時 2025-02-17 10:28:22
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample MLE * 1 -- * 3
other -- * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<vector>
#include<algorithm>
#include<set>
#include<unordered_map>
#include<atcoder/modint>
using namespace std;
using namespace atcoder;
using ll=long long;
using mint=modint998244353;
#define all(v) v.begin(),v.end()
#define rall(v) v.rbegin(),v.rend()
template<class T> bool chmax(T &a, T b){if (a < b){a = b;return true;} else return false;}
template<class T> bool chmin(T &a, T b){if (a > b){a = b;return true;} else return false;}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    ll X;
    cin>>X;
    ll S=0;
    ll ans=0;
    vector<int>vec;
    unordered_map<int,bool>C;
    for(int x=0;x<=300;x++)for(int y=x;y<=300;y++)for(int z=y;z<=300;z++){
        if(C[x*x*x+y*y*y+z*z*z])continue;
        C[x*x*x+y*y*y+z*z*z]=true;
        vec.push_back(x*x*x+y*y*y+z*z*z);
    }
    sort(all(vec));
    vec.erase(unique(all(vec)),vec.end());
    int L=vec.size();
    unordered_map<int,int>mp;
    for(int i=0;i<L;i++)mp[vec[i]]=i;
    vector<vector<int>>V(L),W(L);
    for(int x=0;x<=300;x++)for(int y=x;y<=300;y++)for(int z=y;z<=300;z++){
        V[mp[x*x*x+y*y*y+z*z*z]].push_back(x);
    }
    for(int a=0;S<=X&&a<=300;a++){
        S+=a*a*a;
        for(int b=a;S+b*b*b<=X&&b<=300;b++){
            S+=b*b*b;
            for(int c=b;S+c*c*c<=X&&c<=300;c++){
                S+=c*c*c;
                if(mp.count(X-S)){
                int d=mp[X-S];
                ans+=V[d].end()-lower_bound(all(V[d]),c);
                }
                S-=c*c*c;
            }
            S-=b*b*b;
        }
        S-=a*a*a;
    }
    cout<<ans<<"\n";
}
0