結果

問題 No.1538 引きこもりさんは引き算が得意。
ユーザー kokatsukokatsu
提出日時 2022-11-24 00:37:22
言語 D
(dmd 2.106.1)
結果
WA  
実行時間 -
コード長 1,108 bytes
コンパイル時間 3,568 ms
コンパイル使用メモリ 210,076 KB
実行使用メモリ 5,968 KB
最終ジャッジ日時 2023-09-04 19:01:56
合計ジャッジ時間 4,724 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 WA -
testcase_34 WA -
testcase_35 WA -
testcase_36 WA -
testcase_37 WA -
testcase_38 WA -
testcase_39 WA -
testcase_40 WA -
testcase_41 WA -
testcase_42 WA -
testcase_43 WA -
testcase_44 WA -
testcase_45 WA -
testcase_46 WA -
testcase_47 WA -
testcase_48 WA -
testcase_49 WA -
testcase_50 WA -
testcase_51 WA -
testcase_52 WA -
testcase_53 WA -
testcase_54 WA -
testcase_55 WA -
testcase_56 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std;

void main() {
    long N, K;
    readf("%d %d\n", N, K);

    auto A = readln.chomp.split.to!(long[]);

    long H = N / 2;

    long[long] L, R;
    L[0] = R[0] = 1;

    long l;
    foreach (i; 0 .. H) {
        long[long] nxt;
        nxt[0] = L[0];
        foreach (key, val; L) {
            foreach (j; [-1, 1]) {
                nxt[key+A[i]*j] += val;
            }
        }

        l += A[i];
        L = nxt;
    }

    foreach (a; l == 0 ? [0L] : [-l, 0L, l]) {
        --L[a];
        if (L[a] == 0) L.remove(a);
    }

    long r;
    foreach (i; H .. N) {
        long[long] nxt;
        nxt[0] = R[0];
        foreach (key, val; R) {
            foreach (j; [-1, 1]) {
                nxt[key+A[i]*j] += val;
            }
        }

        r += A[i];
        R = nxt;
    }

    foreach (a; r == 0 ? [0L] : [-r, 0L, r]) {
        --R[a];
        if (R[a] == 0) R.remove(a);
    }

    L.writeln;
    R.writeln;

    bool isOK;
    foreach (key; L.byKey) {
        if (key + K in R) isOK = true;
        if (key - K in R) isOK = true;
    }

    writeln(isOK ? "Yes" : "No");
}
0