結果

問題 No.1538 引きこもりさんは引き算が得意。
ユーザー kokatsu
提出日時 2022-11-24 02:56:15
言語 D
(dmd 2.109.1)
結果
WA  
実行時間 -
コード長 1,281 bytes
コンパイル時間 1,951 ms
コンパイル使用メモリ 209,172 KB
実行使用メモリ 13,452 KB
最終ジャッジ日時 2024-06-22 16:48:28
合計ジャッジ時間 3,445 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 44 WA * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std;

struct S {
    long num, pos, neg;
}

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

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

    long H = N / 2;

    void f(long num, long idx, long lim, ref bool[long][][] list, long pos = 0, long neg = 0) {
        if (idx >= lim) {
            list[pos][neg][num] = true;
        }
        else {
            f(num-A[idx], idx+1, lim, list, pos, 1);
            f(num, idx+1, lim, list, pos, neg);
            f(num+A[idx], idx+1, lim, list, 1, neg);
        }
    }

    auto L = new bool[long][][](2, 2), R = new bool[long][][](2, 2);
    f(0, 0, H, L), f(0, H, N, R);

    bool isOK;

    if (K in R[1][1]) isOK = true;

    foreach (key; L[0][1].byKey) {
        long p = key + K;
        if (p in R[1][0]) isOK = true;
        if (p in R[1][1]) isOK = true;
    }

    foreach (key; L[1][0].byKey) {
        long n = key - K;
        if (n in R[0][1]) isOK = true;
        if (n in R[1][1]) isOK = true;
    }

    foreach (key; L[1][1].byKey) {
        long p = key + K, n = key - K;

        foreach (i; 0 .. 2) {
            foreach (j; 0 .. 2) {
                if (p in R[i][j]) isOK = true;
                if (n in R[i][j]) isOK = true;
            }
        }
    }

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