传送门
明显是一道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;
}