結果

問題 No.1017 Reiwa Sequence
ユーザー knshnbknshnb
提出日時 2020-04-06 18:19:24
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 299 ms / 2,000 ms
コード長 1,885 bytes
コンパイル時間 1,983 ms
コンパイル使用メモリ 207,544 KB
実行使用メモリ 261,688 KB
最終ジャッジ日時 2024-07-03 08:35:21
合計ジャッジ時間 41,148 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 56 ms
85,188 KB
testcase_01 AC 38 ms
61,704 KB
testcase_02 AC 29 ms
50,088 KB
testcase_03 AC 100 ms
131,976 KB
testcase_04 AC 30 ms
49,944 KB
testcase_05 AC 48 ms
73,384 KB
testcase_06 AC 38 ms
61,708 KB
testcase_07 AC 30 ms
49,968 KB
testcase_08 AC 31 ms
49,956 KB
testcase_09 AC 127 ms
167,172 KB
testcase_10 AC 227 ms
249,376 KB
testcase_11 AC 114 ms
155,540 KB
testcase_12 AC 253 ms
260,960 KB
testcase_13 AC 252 ms
260,988 KB
testcase_14 AC 113 ms
155,468 KB
testcase_15 AC 122 ms
167,144 KB
testcase_16 AC 114 ms
155,544 KB
testcase_17 AC 158 ms
190,740 KB
testcase_18 AC 113 ms
155,432 KB
testcase_19 AC 234 ms
249,268 KB
testcase_20 AC 250 ms
249,220 KB
testcase_21 AC 250 ms
249,228 KB
testcase_22 AC 254 ms
249,372 KB
testcase_23 AC 243 ms
249,204 KB
testcase_24 AC 213 ms
249,188 KB
testcase_25 AC 233 ms
249,164 KB
testcase_26 AC 214 ms
249,208 KB
testcase_27 AC 228 ms
249,196 KB
testcase_28 AC 229 ms
249,228 KB
testcase_29 AC 245 ms
249,220 KB
testcase_30 AC 246 ms
249,288 KB
testcase_31 AC 231 ms
249,208 KB
testcase_32 AC 219 ms
249,232 KB
testcase_33 AC 206 ms
249,312 KB
testcase_34 AC 201 ms
249,232 KB
testcase_35 AC 196 ms
249,232 KB
testcase_36 AC 242 ms
249,264 KB
testcase_37 AC 197 ms
249,200 KB
testcase_38 AC 206 ms
249,200 KB
testcase_39 AC 222 ms
249,196 KB
testcase_40 AC 260 ms
261,264 KB
testcase_41 AC 277 ms
261,480 KB
testcase_42 AC 299 ms
261,408 KB
testcase_43 AC 269 ms
261,364 KB
testcase_44 AC 211 ms
261,288 KB
testcase_45 AC 268 ms
261,276 KB
testcase_46 AC 263 ms
261,400 KB
testcase_47 AC 243 ms
261,240 KB
testcase_48 AC 259 ms
261,508 KB
testcase_49 AC 213 ms
261,588 KB
testcase_50 AC 215 ms
261,688 KB
testcase_51 AC 20 ms
38,240 KB
testcase_52 AC 212 ms
249,268 KB
testcase_53 AC 215 ms
249,268 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>  // clang-format off
using Int = int;
#define REP_(i, a_, b_, a, b, ...) for (Int i = (a), lim##i = (b); i < lim##i; i++)
#define REP(i, ...) REP_(i, __VA_ARGS__, __VA_ARGS__, 0, __VA_ARGS__)
struct SetupIO { SetupIO() { std::cin.tie(nullptr), std::ios::sync_with_stdio(false), std::cout << std::fixed << std::setprecision(13); } } setup_io;
#ifndef _MY_DEBUG
#define dump(...)
#endif  // clang-format on

/**
 *    author:  knshnb
 *    created: Mon Apr  6 17:27:48 JST 2020
 **/

template <class T, class S> T make_vec(S x) { return x; }
template <class T, class... Ts> auto make_vec(size_t n, Ts... ts) {
    return std::vector<decltype(make_vec<T>(ts...))>(n, make_vec<T>(ts...));
}

const Int K = 20;
const Int S = K * 150001;
signed main() {
    Int n;
    std::cin >> n;
    std::vector<Int> a(n);
    REP(i, n) std::cin >> a[i];
    Int m = std::min(n, K);
    auto dp = make_vec<short>(m + 1, 2 * S + 1, 0);
    dp[0][S] = 1;
    REP(i, m) {
        auto dp1 = dp[i].begin() + S, dp2 = dp[i + 1].begin() + S;
        REP(j, -S, S) {
            if (dp1[j] == 0) continue;
            dp1[j] = std::min((short)3, dp1[j]);
            dp2[j] += dp1[j];
            dp2[j + a[i]] += dp1[j];
            dp2[j - a[i]] += dp1[j];
        }
    }
    if (dp[m][S] == 1) {
        std::cout << "No" << std::endl;
        return 0;
    }
    Int curj = 0;
    std::vector<Int> b(n);
    for (int i = m - 1; i >= 0; i--) {
        auto dp1 = dp[i].begin() + S, dp2 = dp[i + 1].begin() + S;
        if (dp1[curj - a[i]] > 0) {
            b[i] = a[i];
            curj -= a[i];
        } else if (dp1[curj + a[i]] > 0) {
            b[i] = -a[i];
            curj += a[i];
        } else {
            b[i] = 0;
        }
    }
    assert(curj == 0);
    std::cout << "Yes" << std::endl;
    for (Int x : b) std::cout << x << " ";
    std::cout << std::endl;
}
0