結果

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

ソースコード

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;

    S[] L;
    L ~= S(0, 0, 0);
    foreach (i; 0 .. H) {
        S[] nxt;
        foreach (l; L) {
            foreach (j; -1 .. 2) {
                auto s = l;
                s.num += A[i] * j;
                if (j == -1) s.neg = 1;
                if (j == 1) s.pos = 1;
            }
        }

        L = nxt;
    }

    S[] R;
    R ~= S(0, 0, 0);
    foreach (i; H .. N) {
        S[] nxt;
        foreach (r; R) {
            foreach (j; -1 .. 2) {
                auto s = r;
                s.num += A[i] * j;
                if (j == -1) s.neg = 1;
                if (j == 1) s.pos = 1;
            }
        }

        R = nxt;
    }

    auto X = new long[][][](2, 2);
    foreach (l; L) {
        X[l.pos][l.neg] ~= l.num;
    }

    auto Y = new bool[long][][](2, 2);
    foreach (r; R) {
        Y[r.pos][r.neg][r.num] = true;
    }

    bool isOK;
    foreach (i; 0 .. 4) {
        long p1 = (i >> 1) & 1, n1 = i & 1;

        foreach (j; 0 .. 4) {
            long p2 = (j >> 1) & 1, n2 = j & 1;

            if (p1 == 0 && p2 == 0) continue;
            if (n1 == 0 && n2 == 0) continue;

            foreach (x; X[p1][n1]) {
                if (x in Y[p2][n2]) isOK = true;
            }
        }
    }

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