POJ 3666 Making the Grade (DP)
阿新 • • 發佈:2017-08-16
我們 splay oat amp namespace sort 忽略 clu pre
題意:農夫約翰想修一條盡量平緩的路,路的每一段海拔是A_i,修理後是B_i,花費|A_i – B_i|,求最小花費。
思路:平緩的意思是海拔單調增或單調減(非嚴格),主要目的大概是讓我們做出“到底是增好還是減好”選擇。
用DP的話可以完全忽略這個選擇:
dp[i][j] := 前i + 1個數變成單調且最後一個數是B[j],此時的最小成本
dp[i][j] = min(dp[i – 1][k]) + |A[i] – B[j]| (k = 0,1…j)
#include <cstdio> #include <cstring> #include <algorithm> using namespace std; const int maxn = 2000 + 10; const int inf = 0x3f3f3f3f; int n, cur, pre,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]); for (int i = 1; i < n; i++){ cur = i & 0x1; pre = i-1 & 0x1; int tmp = dp[pre][0]; for (int j = 0; j < n; j++){ tmp = min(tmp, dp[pre][j]); dp[cur][j] = tmp + abs(a[i] - b[j]); } } int ans = inf; for (int i = 0; i < n; i++) ans = min(ans, dp[cur][i]); printf("%d\n", ans); return 0; }
POJ 3666 Making the Grade (DP)