题意就是FJ想要修一条海拔单调增或单调减的路,修之前的海拔是A[i],修之后是B[i],花费|A[i] - B[i]|。求最小花费。
http://poj.org/problem?id=3666
要使花费最小,那么修改后的海拔要么和前一个相等,要么和后一个相等。这样才能保证花费最小。
所以B中的数一定都在A中出现过。
我们考虑对原数组排序,记为b[i]。
dp[i][j]表示:前i + 1个数变成单调序列,最后一个数是b[j] 的最小花费
dp[i][j] = min(dp[i - 1][k]) + abs(a[i] - b[j]), k = 0, 1, 2....j
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 2000 + 10;
const int inf = 0x3f3f3f3f;
int n, a[maxn], b[maxn], dp[maxn][maxn]; //dp[i][j]表示:前i + 1个数变成单调序列,最后一个数是b[j] 的最小花费
int main(){
scanf("%d", &n);
for(int i=0; i<n; i++) scanf("%d", &a[i]);
memcpy(b, a, sizeof(a));
sort(b, b + n);
for(int i=0; i<n; i++) dp[0][i] = abs(a[0] - b[i]);
for(int i=1; i<n; i++){
int tmp = dp[i - 1][0];
for(int j=0; j<n; j++){
tmp = min(tmp, dp[i - 1][j]);
dp[i][j] = tmp + abs(a[i] - b[j]);
}
}
int ans = inf;
for(int i=0; i<n; i++) ans = min(ans, dp[n - 1][i]);
printf("%d\n", ans);
return 0;
}
稍加改动,优化成二维数组O(∩_∩)O~~
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 2000 + 10;
const int inf = 0x3f3f3f3f;
int n, a[maxn], b[maxn], dp[2][maxn]; //dp[i][j]表示:前i + 1个数变成单调序列,最后一个数是b[j] 的最小花费
int main(){
scanf("%d", &n);
for(int i=0; i<n; i++) scanf("%d", &a[i]);
memcpy(b, a, sizeof(a));
sort(b, b + n);
for(int i=0; i<n; i++) dp[0][i] = abs(a[0] - b[i]);
int *pre = dp[0], *nxt = dp[1];
for(int i=1; i<n; i++){
int tmp = pre[0];
for(int j=0; j<n; j++){
tmp = min(tmp, pre[j]);
nxt[j] = tmp + abs(a[i] - b[j]);
}
swap(pre, nxt);
}
int ans = inf;
for(int i=0; i<n; i++) ans = min(ans, pre[i]);
printf("%d\n", ans);
return 0;
}