結果

問題 No.1606 Stuffed Animals Keeper
ユーザー sash0
提出日時 2021-07-23 11:07:08
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 34 ms / 3,000 ms
コード長 3,117 bytes
コンパイル時間 1,965 ms
コンパイル使用メモリ 220,720 KB
最終ジャッジ日時 2025-01-23 07:34:19
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 48
権限があれば一括ダウンロードができます

ソースコード

diff #

// clang-format off
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>

#define mp make_pair
#define fst first
#define snd second
#define forn(i,n) for (int i = 0; i < int(n); i++)
#define forn1(i,n) for (int i = 1; i <= int(n); i++)
#define popcnt __builtin_popcountll
#define ffs __builtin_ffsll
#define ctz __builtin_ctzll
#define clz __builtin_clz
#define clzll __builtin_clzll
#define all(a) (a).begin(), (a).end()

using namespace std;
using namespace __gnu_pbds;

using uint = unsigned int;
using ll = long long;
using ull = unsigned long long;
using pii = pair<int,int>;
using pli = pair<ll,int>;
using pil = pair<int,ll>;
using pll = pair<ll,ll>;
template <typename T> using vec = vector<T>;
using vi = vec<int>;
using vl = vec<ll>;
template <typename T> using que = queue<T>;
template <typename T> using deq = deque<T>;
template <typename T> using ordered_set = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
template <typename K, typename V> using ordered_map = tree<K, V, less<K>, rb_tree_tag, tree_order_statistics_node_update>;

template <typename T> T id(T b) {return b;};
template <typename T> void chmax(T &x, T y) {if (x < y) x = y;}
template <typename T> void chmin(T &x, T y) {if (x > y) x = y;}
template <typename S, typename K> bool contains(S &s, K k) { return s.find(k) != s.end(); }
template <typename T> bool getf(T flag, size_t i) { return (flag>>i) & 1; }
template <typename T> T setf(T flag, size_t i) { return flag | (T(1)<<i); }
template <typename T> T unsetf(T flag, size_t i) { return flag & ~(T(1)<<i); }
void fastio() { ios_base::sync_with_stdio(false); cin.tie(nullptr); }
constexpr ll TEN(int n) { if (n == 0) return 1LL; else return 10LL*TEN(n-1); }
// clang-format on

int main() {
    fastio();

    int n;
    cin >> n;
    vi cnt0, cnt1;
    int c0 = 0, c1 = 0;
    forn(i, n) {
        int x;
        cin >> x;
        if (x == 0)
            c0++;
        else if (x == 1)
            c1++;
        else {
            if (c0 + c1 > 0) {
                cnt0.push_back(c0);
                cnt1.push_back(c1);
            }
            c0 = 0;
            c1 = 0;
        }
    }
    if (c0 + c1 > 0) {
        cnt0.push_back(c0);
        cnt1.push_back(c1);
    }

    const int INF = 11111;
    int base = accumulate(all(cnt1), 0);

    int m = cnt0.size();

    vec<vi> dp(m + 1, vi(n + 1, INF)); // dp[i][j]: i番目までの収支がjのときの最小操作回数
    dp[0][base] = 0;

    forn1(i, m) {
        forn(j, n + 1) {
            if (dp[i - 1][j] == INF) continue;

            // i番目のグループをすべて肉食動物にする
            int k1 = j - cnt1[i - 1];
            chmin(dp[i][k1], dp[i - 1][j] + cnt1[i - 1]);

            // i番目のグループをすべて草食動物にする
            int k2 = j + cnt0[i - 1];
            chmin(dp[i][k2], dp[i - 1][j] + cnt0[i - 1]);
        }
    }

    int ans = 0;
    if (dp[m][base] == INF) {
        ans = -1;
    } else {
        ans = dp[m][base] / 2;
    }

    cout << ans << endl;

    return 0;
}
0