結果
問題 | No.1017 Reiwa Sequence |
ユーザー |
![]() |
提出日時 | 2020-04-03 23:13:17 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 71 ms / 2,000 ms |
コード長 | 1,328 bytes |
コンパイル時間 | 1,806 ms |
コンパイル使用メモリ | 172,372 KB |
実行使用メモリ | 18,176 KB |
最終ジャッジ日時 | 2024-07-03 05:59:37 |
合計ジャッジ時間 | 31,607 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 50 |
ソースコード
#include <bits/stdc++.h> using namespace std; int main() { cin.tie(nullptr); ios::sync_with_stdio(false); int n; cin >> n; vector<int> a(n); for (auto&& e : a) { cin >> e; } vector<int> ord(n); iota(begin(ord), end(ord), 0); sort(begin(ord), end(ord), [&](int i, int j) { return a[i] < a[j]; }); for (int t = 0; t < n - 1; ++t) { if (a[ord[t]] == a[ord[t + 1]]) { cout << "Yes\n"; for (int i = 0; i < n; ++i) { cout << (i == ord[t] ? a[i] : i == ord[t + 1] ? -a[i] : 0) << " \n"[i == n - 1]; } exit(0); } } int m = min(n, 22); vector<int> v(150000 * m); for (int bt = 1; bt < 1 << m; ++bt) { int sum = 0; for (int i = 0; i < m; ++i) { if (bt >> i & 1) { sum += a[i]; } } if (v[sum]) { cout << "Yes\n"; vector<int> res(n); int x = bt & ~v[sum], y = v[sum] & ~bt; assert((x & y) == 0); int s = 0, t = 0; for (int i = 0; i < m; ++i) { if (x >> i & 1) { res[i] = a[i]; s += a[i]; } else if (y >> i & 1) { res[i] = -a[i]; t += a[i]; } } assert(s == t); for (int i = 0; i < n; ++i) { cout << res[i] << " \n"[i == n - 1]; } exit(0); } v[sum] = bt; } cout << "No\n"; }