結果

問題 No.2479 Sum of Squares
ユーザー cureskolcureskol
提出日時 2023-09-23 09:32:02
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 6,150 bytes
コンパイル時間 2,874 ms
コンパイル使用メモリ 216,988 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-07-16 09:15:35
合計ジャッジ時間 4,071 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,944 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 2 ms
6,948 KB
testcase_04 AC 2 ms
6,940 KB
testcase_05 AC 2 ms
6,940 KB
testcase_06 AC 2 ms
6,940 KB
testcase_07 AC 2 ms
6,944 KB
testcase_08 AC 2 ms
6,940 KB
testcase_09 AC 2 ms
6,940 KB
testcase_10 AC 2 ms
6,944 KB
testcase_11 AC 2 ms
6,944 KB
testcase_12 AC 2 ms
6,940 KB
testcase_13 AC 2 ms
6,940 KB
testcase_14 AC 2 ms
6,940 KB
testcase_15 AC 2 ms
6,944 KB
testcase_16 AC 2 ms
6,944 KB
testcase_17 AC 2 ms
6,944 KB
testcase_18 AC 2 ms
6,944 KB
testcase_19 AC 2 ms
6,944 KB
testcase_20 AC 2 ms
6,944 KB
testcase_21 AC 2 ms
6,944 KB
testcase_22 AC 2 ms
6,944 KB
testcase_23 AC 2 ms
6,944 KB
testcase_24 AC 2 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma region template
#pragma GCC optimize("Ofast")
#include <bits/stdc++.h>
using namespace std;

using ll = long long;
using ld = long double;
using vi = vector<int>;
using vll = vector<ll>;
using pi = pair<int, int>;
using pll = pair<ll, ll>;

#define overload2(a, b, c, ...) c
#define overload3(a, b, c, d, ...) d
#define overload4(a, b, c, d, e, ...) e
#define overload5(a, b, c, d, e, f, ...) f

#define TYPE1(T) template <typename T>
#define TYPE2(T, U) template <typename T, typename U>
#define TYPE(...) overload2(__VA_ARGS__, TYPE2, TYPE1)(__VA_ARGS__)
#define TYPES1(T) template <typename... T>
#define TYPES2(H, T) template <typename H, typename... T>
#define TYPES(...) overload2(__VA_ARGS__, TYPES2, TYPES1)(__VA_ARGS__)

#define REP4(i, s, n, d) for (int i = (s); i < (n); i += (d))
#define REP3(i, s, n) REP4(i, s, n, 1)
#define REP2(i, n) REP3(i, 0, n)
#define REP1(n) REP2(tomato, n)
#define REP(...) overload4(__VA_ARGS__, REP4, REP3, REP2, REP1)(__VA_ARGS__)

#define RREP4(i, n, s, d) for (int i = (n)-1; i >= (s); i -= (d))
#define RREP3(i, n, s) RREP4(i, n, s, 1)
#define RREP2(i, n) RREP3(i, n, 0)
#define RREP1(n) RREP2(tomato, n)
#define RREP(...) overload4(__VA_ARGS__, RREP4, RREP3, RREP2, RREP1)(__VA_ARGS__)

#define FOR4(a, b, c, d, v) for (auto [a, b, c, d] : v)
#define FOR3(a, b, c, v) for (auto [a, b, c] : v)
#define FOR2(a, b, v) for (auto [a, b] : v)
#define FOR1(a, v) for (auto a : v)
#define FOR(...) overload5(__VA_ARGS__, FOR4, FOR3, FOR2, FOR1)(__VA_ARGS__)

#define AFOR4(a, b, c, d, v) for (auto &[a, b, c, d] : v)
#define AFOR3(a, b, c, v) for (auto &[a, b, c] : v)
#define AFOR2(a, b, v) for (auto &[a, b] : v)
#define AFOR1(a, v) for (auto &a : v)
#define AFOR(...) overload5(__VA_ARGS__, AFOR4, AFOR3, AFOR2, AFOR1)(__VA_ARGS__)

#define CFOR4(a, b, c, d, v) for (const auto &[a, b, c, d] : v)
#define CFOR3(a, b, c, v) for (const auto &[a, b, c] : v)
#define CFOR2(a, b, v) for (const auto &[a, b] : v)
#define CFOR1(a, v) for (const auto &a : v)
#define CFOR(...) overload5(__VA_ARGS__, CFOR4, CFOR3, CFOR2, CFOR1)(__VA_ARGS__)

#define PFOR(v) for (bool f = true; (f ? exchange(f, false) : next_permutation(ALL(v)));)

#define ALL(v) v.begin(), v.end()
#define RALL(v) v.rbegin(), v.rend()
#define SORT(v) sort(ALL(v))
#define RSORT(v) sort(RALL(v))
#define REVERSE(v) reverse(ALL(v))
#define UNIQUE(v) SORT(v), v.erase(unique(ALL(v)), v.end())
#define SZ(v) int(v.size())
#define MAX(v) *max_element(ALL(v))
#define MIN(v) *min_element(ALL(v))
template <typename T>
conditional_t<is_same_v<T, int>, ll, T> SUM(const vector<T> &v)
{
  return accumulate(ALL(v), conditional_t<is_same_v<T, int>, ll, T>(0));
}

TYPES(T)
void input(T &...a) { (cin >> ... >> a); }
#define DECLARE(T, ...) \
  T __VA_ARGS__;        \
  input(__VA_ARGS__);
#define INT(...) DECLARE(int, __VA_ARGS__)
#define STR(...) DECLARE(string, __VA_ARGS__)
#define LL(...) DECLARE(long long, __VA_ARGS__)
#define CHR(...) DECLARE(char, __VA_ARGS__)
#define DBL(...) DECLARE(double, __VA_ARGS__)
#define VI(n, v) \
  vi v(n);       \
  cin >> v;
#define VLL(n, v) \
  vll v(n);       \
  cin >> v;
#define VS(n, s)       \
  vector<string> s(n); \
  cin >> s;

TYPE(T)
istream &operator>>(istream &is, vector<T> &v)
{
  for (auto &a : v)
    cin >> a;
  return is;
}
TYPE(T)
ostream &operator<<(ostream &os, const vector<T> &v)
{
  if (&os == &cerr)
    os << "[";
  REP(i, v.size())
  {
    os << v[i];
    if (i + 1 < v.size())
      os << (&os == &cerr ? "," : " ");
  }
  if (&os == &cerr)
    os << "]";
  return os;
}
TYPE(T, S)
istream &operator>>(istream &is, pair<T, S> &p)
{
  cin >> p.first >> p.second;
  return is;
}

#ifdef __DEBUG
#include <debug>
#else
#define debug(...) void(0)
#endif

void print()
{
  cout << '\n';
}
TYPES(T, Ts)
void print(const T &a, const Ts &...b)
{
  cout << a;
  (cout << ... << (cout << ' ', b));
  cout << '\n';
}

TYPE(T)
using pq = priority_queue<T>;
TYPE(T)
using pqg = priority_queue<T, vector<T>, greater<T>>;
TYPE(T)
T pick(queue<T> &que)
{
  assert(que.size());
  T a = que.front();
  que.pop();
  return a;
}
TYPE(T)
T pick(pq<T> &que)
{
  assert(que.size());
  T a = que.top();
  que.pop();
  return a;
}
TYPE(T)
T pick(pqg<T> &que)
{
  assert(que.size());
  T a = que.top();
  que.pop();
  return a;
}
TYPE(T)
T pick(stack<T> &sta)
{
  assert(sta.size());
  T a = sta.top();
  sta.pop();
  return a;
}
TYPE(T)
void clear(T &v) { v = decltype(v)(); }

string YES(bool f = true) { return (f ? "YES" : "NO"); }
string Yes(bool f = true) { return (f ? "Yes" : "No"); }
string yes(bool f = true) { return (f ? "yes" : "no"); }

constexpr int INF = 1e9 + 7;
constexpr ll LINF = ll(1e18) + 7;
constexpr ld EPS = 1e-10;

vi iota(int n)
{
  vi a(n);
  iota(ALL(a), 0);
  return a;
}
TYPE(T)
vector<pair<T, int>> query_sort(const vector<T> &v)
{
  vector<pair<T, int>> res(v.size());
  REP(i, v.size())
  res[i] = {v[i], i};
  SORT(res);
  return res;
}
TYPE(T)
void add(vector<T> &v, T a = 1)
{
  AFOR(p, v)
  p += a;
}
TYPE(T)
T rev(T a)
{
  REVERSE(a);
  return a;
}
TYPE(T)
void fin(T a)
{
  cout << a << '\n';
  exit(0);
}
TYPE(T)
bool chmax(T &a, T b) { return (a < b && (a = b, true)); }
TYPE(T)
bool chmin(T &a, T b) { return (a > b && (a = b, true)); }
TYPES(T, Ns)
auto make_vector(T x, int n, Ns... ns)
{
  if constexpr (sizeof...(ns) == 0)
    return vector<T>(n, x);
  else
    return vector(n, make_vector<T>(x, ns...));
}
bool in(const ll S, const int a) { return (S >> a) & 1; }
int popcount(const ll S) { return __builtin_popcountll(S); }
int digit(char c) { return (c >= '0' and c <= '9' ? c - '0' : -1); }
ll sqrtll(ll a)
{
  for (ll b = sqrt(a); b * b <= a; b++)
    if (b * b == a)
      return b;
  for (ll b = sqrt(a); b >= 0 and b * b >= a; b--)
    if (b * b == a)
      return b;
  return -1;
}
#pragma endregion template

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

  LL(s);
  vector<ll> ans;
  while (s)
  {
    ll a = sqrt(s);
    while ((a + 1) * (a + 1) <= s)
      a++;
    while (a * a > s)
      a--;
    ans.push_back(a * a);
    s -= a * a;
  }
  assert(ans.size() <= 15);
  cout << ans.size() << '\n';
  cout << ans << '\n';
}
0