結果

問題 No.1538 引きこもりさんは引き算が得意。
ユーザー kokatsu
提出日時 2022-11-24 00:34:14
言語 D
(dmd 2.109.1)
結果
WA  
実行時間 -
コード長 1,043 bytes
コンパイル時間 2,401 ms
コンパイル使用メモリ 208,636 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-06-22 16:47:26
合計ジャッジ時間 3,626 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 35 WA * 19
権限があれば一括ダウンロードができます

ソースコード

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, 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, r]) {
        --R[a];
        if (R[a] == 0) R.remove(a);
    }

    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