1525D. Armchairs

传送门
明显是一道dp
设dp[i][j] 第i个1 第j个0时 需要付出的最小代价
可以找到递推式
dp[i][j]=min(dp[i][j-1],dp[i-1][j-1]+abs(one[i]-zero[j]));
如果cntp是1的个数
cnt是0的个数
最后的答案就是dp[cntp][cnt]
(比赛时一直在想类似区间dp的写法 应该是漏掉状态了）

//cyc
#pragma GCC optimize("Ofast")
#pragma GCC target("avx,avx2,fma")
#pragma GCC optimization ("unroll-loops")
#include<bits/stdc++.h>
#define rep(i,a,n) for(int i=a;i<=n;i++)
#define per(i,a,n) for(int i=n;i>=a;i--)
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define mst(a) memset(a,0x3f3f3f3f,sizeof a)

using namespace std;
typedef pair<int,int> pii;

int dp[5005][5005];
int one[5005];
int zero[5005];

int main()
{
    ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    int n;
    cin>>n;
    int cnt=0;
    int cntp=0;
    mst(dp);
    int k;
    //cout<<dp[1][1]<<endl;
    for(int i=0;i<=n;i++)dp[0][i]=0;
    for(int i=1;i<=n;i++){
        cin>>k;

        if(!k){
            cnt++;
            zero[cnt]=i;//第x个0位置

        }
        else{
            cntp++;
            one[cntp]=i;//第x个1位置

        }
    }
   for(int i=1;i<=cntp;i++){
    for(int j=1;j<=cnt;j++){
        dp[i][j]=min(dp[i][j-1],dp[i-1][j-1]+abs(one[i]-zero[j]));
    }
   }
   cout<<dp[cntp][cnt]<<endl;
}