結果

問題 No.2510 Six Cube Sum Counting
ユーザー sgfcsgfc
提出日時 2023-10-20 23:32:12
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 3,178 ms / 4,000 ms
コード長 3,894 bytes
コンパイル時間 5,487 ms
コンパイル使用メモリ 319,356 KB
実行使用メモリ 180,280 KB
最終ジャッジ日時 2024-09-20 23:31:18
合計ジャッジ時間 66,619 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2,822 ms
175,284 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2,725 ms
176,692 KB
testcase_07 AC 2,716 ms
174,900 KB
testcase_08 AC 2,733 ms
175,708 KB
testcase_09 AC 2,734 ms
174,768 KB
testcase_10 AC 2,711 ms
174,776 KB
testcase_11 AC 2,717 ms
174,900 KB
testcase_12 AC 2,764 ms
175,028 KB
testcase_13 AC 2,728 ms
174,772 KB
testcase_14 AC 2,709 ms
174,900 KB
testcase_15 AC 2 ms
5,376 KB
testcase_16 AC 2 ms
5,376 KB
testcase_17 AC 2 ms
5,376 KB
testcase_18 AC 2,771 ms
174,772 KB
testcase_19 AC 2,750 ms
174,772 KB
testcase_20 AC 2,487 ms
164,608 KB
testcase_21 AC 3,178 ms
176,688 KB
testcase_22 AC 2,582 ms
164,496 KB
testcase_23 AC 2,569 ms
165,428 KB
testcase_24 AC 2,719 ms
174,388 KB
testcase_25 AC 2,688 ms
174,900 KB
testcase_26 AC 2,813 ms
180,280 KB
testcase_27 AC 2,508 ms
164,624 KB
testcase_28 AC 2,708 ms
174,772 KB
testcase_29 AC 2,641 ms
169,144 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
#include <unordered_map>
#include <unordered_set>

#include <stdio.h>
#include <math.h>
#include <cassert>

#include "atcoder/all"
#include <float.h>

//#include <boost/multiprecision/cpp_int.hpp>
//using namespace boost::multiprecision;
// std::fixed << std::setprecision(10) <<

using namespace std;
using namespace atcoder;

using ll = long long;
using ull = unsigned long long;
using mint = modint998244353;
using mint2 = modint1000000007;

const double eps = 1e-9;

#define REP(i, n) for (ll i = 0; i < ll(n); ++i)
#define REPI(i, init, n) for (ll i = init; i < ll(n); ++i)
#define REPD(i, init) for (ll i = init; i >=0; --i)
#define REPDI(i, init, n) for (ll i = init; i >=n; --i)
#define REPE(i, c) for (const auto& i : c)
#define TCASE() ll _ttt; cin >> _ttt; while(_ttt--)

using vl = vector<ll>;
#define VL(a, n) vector<ll> a; a.assign(n, 0);
#define VLI(a, n) vector<ll> a; a.assign(n, 0); for(auto& x : a) cin >> x;
#define VSI(a, n) vector<string> a; a.assign(n, ""); for(auto& x : a) cin >> x;
using vvl = vector<vector<ll>>;
#define VVLI(a, n, m, init) vector<vector<ll>> a(n); for(auto& x : a) x.assign(m, init);
using vd = vector<double>;
using pl = pair<ll, ll>;

struct uv { ll u; ll v; ll c; };
#define VUVI(a, n) vector<uv> a; a.assign(n, {0, 0, 1}); for(auto& x : a) {cin >> x.u >> x.v; x.u--; x.v--;}
#define VUVCI(a, n) vector<uv> a; a.assign(n, {0, 0, 0}); for(auto& x : a) {cin >> x.u >> x.v >> x.c; x.u--; x.v--;}
vvl to_edge(const ll n, const vector<uv>& v) { vvl ret(n); for (auto& x : v) ret[x.u].push_back(x.v); return ret; }
vvl to_edge_d(const ll n, const vector<uv>& v) { vvl ret(n); for (auto& x : v) { ret[x.u].push_back(x.v); ret[x.v].push_back(x.u); } return ret; }

template <class T = long long> using pql = priority_queue<T>;
template <class T = long long> using pqg = priority_queue<T, vector<T>, greater<T>>;
using vm = vector<mint>;
using vvm = vector<vm>;

void yn(bool f) { std::cout << (f ? "Yes" : "No") << endl; };
template<class T> void ov(const T& v) { for (auto it = v.begin(); it != v.end(); it++) { if (it != v.begin()) cout << " "; cout << *it; } };
template<> void ov(const vm& v) { for (auto it = v.begin(); it != v.end(); it++) { if (it != v.begin()) cout << " "; cout << it->val(); } };

const ll llhuge = 1LL << 60; //十分でかいが多少足しても溢れない数値
auto gmax(const auto& a, const auto& b) { return a > b ? a : b; };
auto gmin(const auto& a, const auto& b) { return a < b ? a : b; };
template<class T, class U> bool chmin(T& a, const U& b) { const T olda = a; a = gmin(a, b); return olda != a; }
template<class T, class U> bool chmax(T& a, const U& b) { const T olda = a; a = gmax(a, b); return olda != a; }
template<class T> void mysort(T& v) { std::sort(begin(v), end(v)); };
template<class T, class U> void mysort(T& v, U pr) { std::sort(begin(v), end(v), pr); };
template<class T> void myrev(T& v) { std::reverse(begin(v), end(v)); };

int main() {
	cin.tie(nullptr);
	ios::sync_with_stdio(false);

	ll x;
	cin >> x;

	ll emin = 0;
	REP(i, 301) {
		if ((x - 300 * 300 * 300)/5 > i * i * i) emin = i;
		else break;
	}
	ll dmin = 0;
	REP(i, 301) {
		if ((x - 300 * 300 * 300 * 2)/4 > i * i * i) dmin = i;
		else break;
	}
	ll dmax = 0;
	REP(i, 301) {
		if ((x - i * i * i * 3) > 0) dmax = i;
		else break;
	}

	unordered_map<ll, ll> mp;
	REP(c, dmin) {
		REP(b, c + 1) REP(a, b + 1) {
			ll tx = a * a * a + b * b * b + c * c * c;
			mp[tx]++;
		}
	}

	ll ans = 0;
	REPI(d, dmin, dmax + 1) {
		ll c = d;
		REP(b, c + 1) REP(a, b + 1) {
			ll tx = a * a * a + b * b * b + c * c * c;
			mp[tx]++;
		}

		ll emax = 300;
		REP(i, 301) {
			if ((x - d * d * d) > i * i * i * 2) emax = i;
			else break;
		}

		REPI(e, max(d, emin), emax + 1) REPI(f, e, 301) {
			const ll t = x - (d * d * d + e * e * e + f * f * f);
			if (t < 0) continue;
			ans += mp[t];
		}
	}

	cout << ans << endl;
}
0