結果

問題 No.1606 Stuffed Animals Keeper
ユーザー ktr216
提出日時 2021-07-16 22:53:42
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 31 ms / 3,000 ms
コード長 1,127 bytes
コンパイル時間 1,387 ms
コンパイル使用メモリ 172,232 KB
実行使用メモリ 35,968 KB
最終ジャッジ日時 2024-07-06 10:25:47
合計ジャッジ時間 2,780 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 48
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#define rep(i, l, r) for (int i = (l); i < (r); i++)
using namespace std;

typedef long long ll;

int main() {
    int N;
    cin >> N;
    vector<int> A(N, 0), B(1, 0), C(3, 0), D(1, 0);
    rep(i, 0, N) {
        cin >> A[i];
        C[A[i]]++;
        if (A[i] == 2) {
            if (B.back() > 0) {
                B.push_back(0);
                D.push_back(0);
            }
        } else {
            B.back()++;
            D.back() += A[i];
        }
    }
    if (B.back() == 0) {
        B.pop_back();
        D.pop_back();
    }
    int M = B.size();
    //rep(i, 0, M) cout << B[i] << " "; cout << endl;
    //rep(i, 0, M) cout << D[i] << " "; cout << endl;
    vector<vector<int>> dp(M + 1, vector<int>(N + 1, 1e9));
    dp[0][0] = 0;
    rep(i, 0, M) {
        rep(j, 0, N) {
            if (dp[i][j] == 1e9) continue;
            dp[i + 1][j] = min(dp[i + 1][j], dp[i][j] + D[i]);
            dp[i + 1][j + B[i]] = min(dp[i + 1][j + B[i]], dp[i][j] + B[i] - D[i]);
        }
    }
    int ans = dp[M][C[1]];
    if (ans == 1e9) ans = -1;
    else ans /= 2;
    cout << ans << endl;
}
0