D. Armchairs

题目链接

题意：给出一些座位，1表示有人坐，0表示空位置，假设一个人从原来的座位i移动到空座位j上，那么花费是abs(i-j)，问把最初有座位的人全部移到最初是空的位置上（一个位置只能坐一个人），最小总花费是多少

题解：

代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<queue>
#include<stack>
#include<map>
#include<cmath>
#include<cstdlib>
#include<string>
#include<vector>
using namespace std;
#define scd(n) scanf("%d",&n)
#define scdd(a,b) scanf("%d%d",&a,&b)
#define lowbit(x) x&(-x)
typedef long long LL;
typedef pair<int,int>PII;
const int N = 5000+10;
const int INF = 0x3f3f3f3f;
int one[N],zero[N];
int dp[N][N];

void solve()
{
    int n;
    int cnt1=0,cnt2=0;
    scd(n);
    for(int i=1;i<=n;i++){
        int x;
        scd(x);
        if(x==1) one[++cnt1]=i;
        else zero[++cnt2]=i;
    }
   memset(dp,0x3f,sizeof dp);
   for(int i=0;i<=cnt2;i++) dp[0][i]=0;
    for(int i=1;i<=cnt1;i++){
        for(int j=1;j<=cnt2;j++){
            dp[i][j]=min(dp[i][j-1],dp[i-1][j-1]+abs(one[i]-zero[j]));
        }
    }
    cout << dp[cnt1][cnt2] << endl;
}

int main()
{
    solve();
    return 0;
}