結果

問題 No.1017 Reiwa Sequence
ユーザー ir5
提出日時 2024-06-14 22:10:29
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,220 bytes
コンパイル時間 1,590 ms
コンパイル使用メモリ 143,096 KB
実行使用メモリ 13,884 KB
最終ジャッジ日時 2024-06-14 22:11:07
合計ジャッジ時間 18,837 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 WA * 3
other AC * 5 WA * 10 TLE * 1 -- * 34
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <cassert>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <iostream>
#include <tuple>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <ranges>
#include <iomanip>
using namespace std;
using ll = long long;

auto range(int n) { return views::iota(0, n); }

template<class T, class U> ostream& operator<<(ostream& os, const pair<T, U>& p){ return os << "{" << p.first << ", " << p.second << "}"; }
template<typename T> ostream& operator<<(ostream& os, const vector<T>& obj) { os << "{"; for (const auto& e : obj) os << e << ", "; return os << "}"; }
template<typename T> ostream& operator<<(ostream& os, const set<T>& obj) { os << "set{"; for (const auto& e : obj) os << e << ", "; return os << "}"; }
template<typename T, typename U> ostream& operator<<(ostream& os, const map<T, U>& obj) { os << "map{"; for (const auto& e : obj) os << e << ", "; return os << "}"; }

#ifdef ONLINE_JUDGE
#define dump(expr) ;
#else
#define dump(expr) { cerr << "\033[33m#L" << __LINE__ << ": " << expr << "\033[39m" << endl; }
#endif

void solve() {
  int n; cin >> n;
  vector<ll> vs(n);
  for (int i : range(n)) cin >> vs[i];

  int dmax = min(n, 19);
  vector<char> sign(dmax);
  int tot = 0;
  while (1) {
    if (sign[0] == 1) tot -= vs[0];
    else if (sign[0] == 2) tot -= -vs[0];
    sign[0]++;
    if (sign[0] == 1) tot += vs[0];
    else if (sign[0] == 2) tot += -vs[0];

    {
      int j = 0;
      for (; j < dmax && sign[j] == 3; j++) {
        sign[j] = 0;

        if (sign[j + 1] == 1) tot -= vs[j + 1];
        else if (sign[j + 1] == 2) tot -= -vs[j + 1];
        sign[j + 1]++;
        if (sign[j + 1] == 1) tot += vs[j + 1];
        else if (sign[j + 1] == 2) tot += -vs[j + 1];
      }
      if (sign[dmax - 1] == 2) break;
    }

    if (tot == 0) {
      for (int j : range(dmax)) {
        if (j) cout << " ";

        if (sign[j] == 0) cout << 0;
        if (sign[j] == 1) cout << vs[j];
        if (sign[j] == 2) cout << -vs[j];
      }
      for (int j = dmax; j < n; ++j) cout << " 0";
      cout << endl;
      return;
    }
  }
  cout << "No" << endl;
}

int main() {
  cout << fixed << setprecision(12);
  solve();
}
0