結果

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

ソースコード

diff #

#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define ALL(v) v.begin(), v.end()
typedef long long ll;

#include <bits/stdc++.h>
using namespace std;

int dp[5010][5010];
const int INF=1e9+7;

int main(){
  int n;
  cin>>n;
  
  vector<int> A(n+1);
  rep(i,n) cin>>A[i];
  A[n]=2;
  
  vector<int> B,C,D;
  int cnt0=0,cnt1=0,sum0=0;
  rep(i,n+1){
    if(A[i]==0){
      cnt0++;
      sum0++;
    }
    else if(A[i]==1) cnt1++;
    else{
      if(cnt0==0 && cnt1==0) continue;
      else{
        B.push_back(cnt0);
        C.push_back(cnt1);
        D.push_back(cnt0+cnt1);
        cnt0=0,cnt1=0;
      }
    }
  }
  
  rep(i,5010) rep(j,5010) dp[i][j]=INF;
  dp[0][sum0]=0;
  int m=B.size();
  rep(i,m){
    rep(j,sum0+1){
      if(dp[i][j]==INF) continue;
      dp[i+1][j]=min(dp[i+1][j],dp[i][j]+B[i]);
      if(j-D[i]>=0) dp[i+1][j-D[i]]=min(dp[i+1][j-D[i]],dp[i][j]+C[i]);
    }
  }
  
  if(dp[m][0]==INF) cout<<-1<<endl;
  else cout<<dp[m][0]/2<<endl;

  return 0;
}
0