結果

問題 No.1606 Stuffed Animals Keeper
ユーザー nebukuro09
提出日時 2021-07-16 22:53:33
言語 D
(dmd 2.109.1)
結果
AC  
実行時間 102 ms / 3,000 ms
コード長 1,406 bytes
コンパイル時間 807 ms
コンパイル使用メモリ 134,676 KB
実行使用メモリ 70,908 KB
最終ジャッジ日時 2024-06-22 11:46:56
合計ジャッジ時間 3,328 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 48
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std.stdio, std.array, std.string, std.conv, std.algorithm;
import std.typecons, std.range, std.random, std.math, std.container;
import std.numeric, std.bigint, core.bitop, core.stdc.stdlib, std.datetime;


void main() {
    auto N = readln.chomp.to!int;
    auto A = readln.split.map!(to!int).array;
    auto B = A.split(2);
    int[][] C;

    foreach (b; B) {
        if (b.empty) continue;
        C ~= new int[](2);
        C.back[0] = b.count(0).to!int;
        C.back[1] = b.count(1).to!int;
    }

    auto M = C.length.to!int;

    auto dp = new int[][](M+1, N*2);
    foreach (i; 0..M+1) dp[i][] = 1 << 29;
    dp[0][0] = 0;

    int zero_sum = 0;
    int one_sum = 0;

    foreach (i; 0..M) {
        int zero = C[i][0];
        int one = C[i][1];
        foreach (zero_used; 0..N+1) {
            int one_used = zero_sum + one_sum - zero_used;
            int zero_amari = max(0, zero_sum - zero_used);
            dp[i+1][zero_used+zero+one] = min(dp[i+1][zero_used+zero+one], dp[i][zero_used] + max(0, one - zero_amari));
            int one_amari = max(0, one_sum - one_used);
            dp[i+1][zero_used] = min(dp[i+1][zero_used], dp[i][zero_used] + max(0, zero - one_amari));
        }
        zero_sum += C[i][0];
        one_sum += C[i][1];
    }

    auto ans = dp[M][A.count(0).to!int];
    if (ans == 1 << 29) {
        writeln(-1);
    } else {
        writeln(ans);
    }
}
0