poj3666(dp前缀优化)

链接：点击打开链接

题意：将A1....An变为B1.....Bn,要求序列B满足非严格单调递增或非严格单调递减，代价为|A₁-B₁|+|A_2-B₂|+...+|A_N -B_N|，输出最小代价

代码：

#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;
}