Educational Codeforces Round 109（D. Armchairs）

计算将所有1与0匹配的最小距离和

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pii pair<int,int>
#define inf 0x3f3f3f3f
#define rush() int T;cin>>T;while(T--)
#define pb push_back
#define mp make_pair
#define ms(a,x) memset(a,x,sizeof(a))
#define yes(); cout<<"YES\n";
#define no(); cout<<"NO\n";
const int MAX=3e5+5;
int pos1[2505];
int pos0[5005];
int dp[2505][5005];//dp[i][j]:前i个1与前j个0匹配的最小花费
int main(){
    int n,t1=0,t0=0;
    cin>>n;
    for(int i=1;i<=n;i++){
        int x;
        cin>>x;
        if(x&1)pos1[++t1]=i;
        else pos0[++t0]=i;
    }
    ms(dp,inf);
    ms(dp[0],0);//一个1也没匹配，不增加操作距离
    //思想：最长公共子序列
    //变化：单个序列里，某些字符之间的匹配
    //		以某个字符为主，将它与另一个字符匹配
    //限制：1的个数不超过n/2，如果超过，有剩余的1不能匹配
    for(int i=1;i<=t1;i++)
        for(int j=i;j<=t0;j++)//  1不和当前0匹配，1和之前0匹配      匹配，加距离
            dp[i][j]=min(dp[i][j-1],dp[i-1][j-1]+abs(pos1[i]-pos0[j]));
    cout<<dp[t1][t0]<<endl;
    //    for(int i=1;i<=t1;i++){
//        for(int j=1;j<=t0;j++)
//            cout<<dp[i][j]<<"\t";
//        cout<<endl;
//    }
	//优化处理j从i开始：j<i的状态均为无穷大，dp[2][1]=min(dp[2][0],dp[1][0]+dis),转移到dp[2][1]的状态均无效
    //样例
    //1 1 1 0 0 0
    //dp[ij]值
    //11 3 12 3 13 3
    //21 x 22 6 23 6
    //31 x 32 x 33 9
    //x:inf
}