首页 > 其他分享 >D. Armchairs

D. Armchairs

时间:2024-06-18 16:23:14浏览次数:26  
标签:min int ren cnt1 ans Armchairs dp

原题链接

题解

1.改变座位之后,保持人的相对顺序不变一定使答案不劣
2.\(n\) 不是很大,因此可以考虑 \(O(n^2)\) 的做法
3.令 \(dp[i][j]\) 为第 \(i\) 个人移到位置 \(j\),且 \([1,i-1]\) 的人都已经移到了最优位置时的最小花费,\(index[i]\) 为第 \(i\) 个人的下标
则 \(dp[i][j]=\min(dp[i-1][k])+abs(j-index[i])\),其中 \(k\in[1,j-1]\) 且 \(a[k]=0\)
可以线性维护

code

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

int dp[5005][5005]={0};
int a[5005];
int main()
{
    int n;
    cin>>n;

    vector<int> ren;
    ren.push_back(0);
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        if(a[i]) ren.push_back(i);
    }


    int cnt1=ren.size()-1,ans=(cnt1?2e9:0);
    for(int i=1;i<=cnt1;i++)
    {
        int cnt0=0;
        int mn;
        for(int j=1;j<=n;j++)
        {
            if(a[j]) continue;

            cnt0++;
            if(cnt0==i) mn=dp[i-1][cnt0-1];
            else if(cnt0>i) mn=min(mn,dp[i-1][cnt0-1]);
            else continue;

            dp[i][cnt0]=mn+abs(ren[i]-j);
            if(i==cnt1) ans=min(ans,dp[i][cnt0]);
        }
    }

    cout<<ans;
    return 0;
}

标签:min,int,ren,cnt1,ans,Armchairs,dp
From: https://www.cnblogs.com/pure4knowledge/p/18254586

相关文章