結果

問題 No.1017 Reiwa Sequence
ユーザー ir5ir5
提出日時 2024-06-14 22:37:24
言語 C++23(gcc13)
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 99 ms / 2,000 ms
コード長 2,507 bytes
コンパイル時間 2,533 ms
コンパイル使用メモリ 160,784 KB
実行使用メモリ 9,472 KB
最終ジャッジ日時 2024-06-14 22:38:03
合計ジャッジ時間 38,248 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 2 ms
6,944 KB
testcase_03 AC 2 ms
6,940 KB
testcase_04 AC 2 ms
6,944 KB
testcase_05 AC 2 ms
6,944 KB
testcase_06 AC 2 ms
6,940 KB
testcase_07 AC 2 ms
6,944 KB
testcase_08 AC 2 ms
6,944 KB
testcase_09 AC 2 ms
6,944 KB
testcase_10 AC 29 ms
7,168 KB
testcase_11 AC 2 ms
6,940 KB
testcase_12 AC 47 ms
9,088 KB
testcase_13 AC 48 ms
8,960 KB
testcase_14 AC 3 ms
6,940 KB
testcase_15 AC 3 ms
6,940 KB
testcase_16 AC 3 ms
6,944 KB
testcase_17 AC 4 ms
6,940 KB
testcase_18 AC 2 ms
6,940 KB
testcase_19 AC 27 ms
6,944 KB
testcase_20 AC 28 ms
6,944 KB
testcase_21 AC 24 ms
6,940 KB
testcase_22 AC 28 ms
6,940 KB
testcase_23 AC 27 ms
6,940 KB
testcase_24 AC 28 ms
7,168 KB
testcase_25 AC 26 ms
6,940 KB
testcase_26 AC 23 ms
6,944 KB
testcase_27 AC 25 ms
6,940 KB
testcase_28 AC 25 ms
6,940 KB
testcase_29 AC 27 ms
6,940 KB
testcase_30 AC 28 ms
7,168 KB
testcase_31 AC 22 ms
6,940 KB
testcase_32 AC 22 ms
6,940 KB
testcase_33 AC 23 ms
6,940 KB
testcase_34 AC 16 ms
6,940 KB
testcase_35 AC 17 ms
6,940 KB
testcase_36 AC 23 ms
6,944 KB
testcase_37 AC 16 ms
6,940 KB
testcase_38 AC 17 ms
6,940 KB
testcase_39 AC 26 ms
6,940 KB
testcase_40 AC 76 ms
8,960 KB
testcase_41 AC 67 ms
7,504 KB
testcase_42 AC 67 ms
7,808 KB
testcase_43 AC 71 ms
8,588 KB
testcase_44 AC 49 ms
6,944 KB
testcase_45 AC 72 ms
8,192 KB
testcase_46 AC 67 ms
7,680 KB
testcase_47 AC 67 ms
7,808 KB
testcase_48 AC 99 ms
9,472 KB
testcase_49 AC 66 ms
6,940 KB
testcase_50 AC 66 ms
6,944 KB
testcase_51 AC 2 ms
6,944 KB
testcase_52 AC 23 ms
6,944 KB
testcase_53 AC 17 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC optimize "Ofast"
#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

map<int, int> brute(vector<int> vs) {
  int n = vs.size();
  int loop = pow(3, n);
  map<int, int> st;
  for (int b : range(loop)) {
    int tot = 0;
    for (int j = 0, y = b; y; y /= 3, j++) {
      if (y % 3 == 1) tot += vs[j];
      if (y % 3 == 2) tot -= vs[j];
    }
    st[tot] = b;
  }
  return st;
}

void recover(vector<int> vs, int b, vector<int>& sign) {
  int n = vs.size();
  sign.resize(n);
  for (int j = 0, y = b; j < n; y /= 3, j++) {
    if (y % 3 == 0) sign[j] = 0;
    if (y % 3 == 1) sign[j] = vs[j];
    if (y % 3 == 2) sign[j] = -vs[j];
  }
}

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

  int dmax = min(n, 20);
  int half = dmax / 2;

  vector<int> v1(vs.begin(), vs.begin() + half);
  vector<int> v2(vs.begin() + half, vs.begin() + dmax);

  auto st1 = brute(v1);
  auto st2 = brute(v2);

  for (auto p : st1) if (st2.count(-p.first)) {
    auto q = *st2.find(-p.first);
    if (p.second == 0 && q.second == 0) continue;

    cout << "Yes" << endl;
    vector<int> sign;
    recover(v1, p.second, sign);
    for (int s : sign) cout << s << " ";

    recover(v2, q.second, sign);
    for (int s : sign) cout << s << " ";

    for (int j = dmax; j < n; ++j) cout << "0 ";
    cout << endl;
    return;
  }
  cout << "No" << endl;
}

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