AGC032D Rotation Sort

AGC032D

有个排列\(p_i\),每次可以以\(A\)的代价将某个数移到右边的一个位置,或者可以以\(B\)的代价移到左边的一个位置。

问将其变成升序排列花的最小代价。

\(n\le 5*10^3\)


考虑\(p_x=1\)的位置:最终它一定要移到最左边。移到最左边有两种方法:花\(B\)的代价将\(p_x\)移到最左边;花\(A*左边的数的个数\)的代价将左边的数移到\(p_x\)的右边。注意到如果我们用了后面的这个操作,我们完全可以让\(p_x\)左边的数任意插入到\(x\)右边的任意位置,于是可以先将\(x\)右边的数排列好,直接按照顺序插进去一定是最优的。

第一种操作相当于把序列中的\(1\)剔除掉,接着做剩下的子问题;第二种操作形成的子问题为原来的一段后缀。

于是可以DP:\(f_{i,j}\)表示考虑区间\([i,n]\),剔除掉了小于等于\(j\)的数,这个问题的最小代价。

转移的时候需要一个辅助数组\(c_{i,j}\)表示\([j,i]\)中大于\(p_j\)的数的个数。

总时间复杂度\(O(n^2)\)


using namespace std;
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <climits>
#define N 5010
#define ll long long
int n;
ll A,B;
int p[N],re[N];
ll f[N][N];
int c[N][N];
void upd(ll &a,ll b){a=min(a,b);}
int main(){
//	freopen("in.txt","r",stdin);
	scanf("%d%lld%lld",&n,&A,&B);
	for (int i=1;i<=n;++i)
		scanf("%d",&p[i]),re[p[i]]=i;
	for (int j=1;j<=n;++j)
		for (int i=re[j]-1;i>=1;--i)
			c[j][i]=c[j][i+1]+(p[i]>j);
	memset(f,127,sizeof f);
	for (int j=0;j<=n;++j)
		f[n+1][j]=0;
	for (int i=n;i>=1;--i){
		f[i][n]=0;
		for (int j=n-1;j>=0;--j){
			if (c[j+1][i]==0)
				f[i][j]=f[i][j+1];
			else
				f[i][j]=min(f[i][j+1]+B,f[re[j+1]+1][j+1]+A*c[j+1][i]);
		}
	}
	printf("%lld\n",f[1][0]);
	return 0;
}
posted @ 2020-09-28 22:11  jz_597  阅读(120)  评论(0编辑  收藏  举报