結果
問題 | No.2510 Six Cube Sum Counting |
ユーザー |
|
提出日時 | 2023-10-20 22:45:31 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
MLE
|
実行時間 | - |
コード長 | 1,480 bytes |
コンパイル時間 | 978 ms |
コンパイル使用メモリ | 98,008 KB |
最終ジャッジ日時 | 2025-02-17 10:21:11 |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | TLE * 1 MLE * 2 -- * 1 |
other | -- * 26 |
ソースコード
#include<iostream>#include<vector>#include<algorithm>#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;for(int x=0;x<=300;x++)for(int y=x;y<=300;y++)for(int z=y;z<=300;z++){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";}