poj3666(dp前缀优化)

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