結果

問題 No.2510 Six Cube Sum Counting
コンテスト
ユーザー Misuki
提出日時 2024-01-17 01:09:33
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
MLE  
実行時間 -
コード長 2,568 bytes
コンパイル時間 2,240 ms
コンパイル使用メモリ 204,356 KB
実行使用メモリ 596,096 KB
最終ジャッジ日時 2024-09-28 03:06:39
合計ジャッジ時間 12,836 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample MLE * 1 -- * 3
other -- * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC optimize("O2")
#include <algorithm>
#include <array>
#include <bit>
#include <bitset>
#include <cassert>
#include <cctype>
#include <cfenv>
#include <cfloat>
#include <chrono>
#include <cinttypes>
#include <climits>
#include <cmath>
#include <compare>
#include <complex>
#include <concepts>
#include <cstdarg>
#include <cstddef>
#include <cstdint>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <deque>
#include <fstream>
#include <functional>
#include <initializer_list>
#include <iomanip>
#include <ios>
#include <iostream>
#include <istream>
#include <iterator>
#include <limits>
#include <list>
#include <map>
#include <memory>
#include <new>
#include <numbers>
#include <numeric>
#include <ostream>
#include <queue>
#include <random>
#include <ranges>
#include <set>
#include <span>
#include <sstream>
#include <stack>
#include <streambuf>
#include <string>
#include <tuple>
#include <type_traits>
#include <variant>

//#define int ll
#define INT128_MAX (__int128)(((unsigned __int128) 1 << ((sizeof(__int128) * __CHAR_BIT__) - 1)) - 1)
#define INT128_MIN (-INT128_MAX - 1)

#ifdef DEBUG
#define dbg(x) cout << (#x) << " = " << x << '\n'
#else
#define dbg(x)
#endif

namespace R = std::ranges;
namespace V = std::views;

using namespace std;

using ll = long long;
using ull = unsigned long long;
using ldb = long double;
//#define double ldb

template<class T, size_t N>
ostream& operator<<(ostream& os, const array<T, N> &arr) {
  for(const T &X : arr)
    os << X << ' ';
  return os;
}
template<class T>
ostream& operator<<(ostream& os, const vector<T> &vec) {
  for(const T &X : vec)
    os << X << ' ';
  return os;
}
template<class T>
ostream& operator<<(ostream& os, const set<T> &s) {
  for(const T &x : s)
    os << x << ' ';
  return os;
}

map<int, int> m[301]; //{{c, sum}, cnt}
void dfs(int i, int pre, int sum) {
  if (i == 3) {
    m[pre][sum]++;
  } else {
    for(int j = pre; j <= 300; j++)
      dfs(i + 1, j, sum + j * j * j);
  }
}

map<int, int> m2[301];
void dfs2(int i, int pre, int sum) {
  if (i == 2) {
    m2[pre][sum]++;
  } else {
    for(int j = 0; j <= pre; j++)
      dfs2(i - 1, j, sum + j * j * j);
  }
}

signed main() {
  ios::sync_with_stdio(false), cin.tie(NULL);

  dfs(0, 0, 0);
  dfs2(5, 300, 0);

  int x; cin >> x;

  ll ans = 0;
  map<int, int> tmp;
  for(int c = 300; c >= 0; c--) {
    for(auto [sum, cnt] : m2[c])
      tmp[sum] += cnt;
    for(auto [sum, cnt] : m[c])
      if (tmp.contains(x - sum))
        ans += (ll)cnt * tmp[x - sum];
  }

  cout << ans << '\n';

  return 0;
}
0