链接:点击打开链接
题意:将A1....An变为B1.....Bn,要求序列B满足非严格单调递增或非严格单调递减,代价为|A1-B1|+|A2-B2|+...+|AN -BN|,输出最小代价
代码:
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
const int INF=0x3f3f3f3f;
int a[2005],b[2005],dp[2005][2005];
int main(){ //dp[i][j]表示第i个数高度为第j个时的代价
int n,i,j,ans,tmp;
while(scanf("%d",&n)!=EOF){
for(i=0;i<n;i++){
scanf("%d",&a[i]);
b[i]=a[i];
} //变化后递增
sort(b,b+n); //最优情况变换后的值一定出现在原来的数列中
for(i=0;i<n;i++)
dp[0][i]=abs(a[0]-b[i]);
for(i=1;i<n;i++){
tmp=INF;
for(j=0;j<n;j++){
tmp=min(tmp,dp[i-1][j]); //高度j之前的最小值,减少了一次循环
dp[i][j]=tmp+abs(a[i]-b[j]);
}
}
ans=INF;
for(i=0;i<n;i++)
ans=min(ans,dp[n-1][i]);
sort(b,b+n,greater<int>()); //变换后递减
for(i=0;i<n;i++)
dp[0][i]=abs(a[0]-b[i]);
for(i=1;i<n;i++){
tmp=INF;
for(j=0;j<n;j++){
tmp=min(tmp,dp[i-1][j]);
dp[i][j]=tmp+abs(a[i]-b[j]);
}
}
for(i=0;i<n;i++)
ans=min(ans,dp[n-1][i]);
printf("%d\n",ans);
}
return 0;
}