题意
略
思路
杰哥提醒的dp,一开始想的就是dp,但状态就是定义不出来囧。
状态定义:dp[i][j]:前i个数已经排好,第i个数是第j小的最小所需要改变的值。
状态转移:dp[i][j] = min(dp[i-1][1...j]+abs(a[i]-b[j]));
需要用到离散化,b数组就是保存离散化后的值,a数组是原来的值。
2857数据比较弱(数组大小2005),3698数据增强了(数组大小5005以及一些细节)。
code
#include <bits/stdc++.h>
using namespace std;
const int N = 5005;
typedef long long LL;
const LL INF = 0x3f3f3f3f3f3f3f3f;
int n, tn;
int a[N], b[N];
LL dp[2][N];
LL pre[N];
int main() {
while(scanf("%d", &n) != EOF) {
if(n <= 1) {
puts("0");
continue;
}
for(int i = 1;i <= n; i++) {
scanf("%d" ,&a[i]);
b[i] = a[i];
}
sort(b+1, b+n+1);
tn = unique(b+1, b+n+1)-(b+1);
memset(dp, INF, sizeof(dp));
int cur = 0;
for(int i = 1;i <= tn; i++) dp[0][i] = abs(b[i]-a[1]);
for(int i = 2;i <= n; i++) {
pre[0] = INF;
for(int j = 1;j <= tn; j++)
pre[j] = min(pre[j-1], dp[cur][j]);
for(int j = 1;j <= tn; j++) {
int val = abs(b[j]-a[i]);
dp[cur^1][j] = pre[j]+val;
}
cur ^= 1;
}
LL res = INF;
for(int i = 1;i <= tn; i++)
res = min(res, dp[cur][i]);
printf("%lld\n", res);
}
return 0;
}