1. 程式人生 > >POJ 3666 Making the Grade (DP)

POJ 3666 Making the Grade (DP)

我們 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)