题目来源:
HackerRank
基准时间限制:1 秒 空间限制:131072 KB 分值: 20
难度:3级算法题
数组A包含N个元素A1, A2......AN。数组B包含N个元素B1, B2......BN。并且数组A中的每一个元素Ai,都满足1 <= Ai <= Bi。数组A的代价定义如下:
(公式表示所有两个相邻元素的差的绝对值之和)
给出数组B,计算可能的最大代价S。
Input
第1行:1个数N,表示数组的长度(1 <= N <= 50000)。 第2 - N+1行:每行1个数,对应数组元素Bi(1 <= Bi <= 10000)。
Output
输出最大代价S。
Input示例
5 10 1 10 1 10
Output示例
36
既然是找最大代价,而且每两个点之间的最大代价只与这两点有关,每个点要么取峰值,要么取1;
这就类似是一个01背包;每个点有两种状态,一是取峰值,二是取1;
动态方程如下:
dp[i][0]=max(dp[i-1][0], dp[i-1][1]+abs(b[i-1]-1));
dp[i][1]=max(dp[i-1][0]+abs(b[i]-1), dp[i-1][1]+abs(b[i]-b[i-1]));
#include <iostream>
#include <algorithm>
#include <queue>
#include <cstring>
#include <cstdio>
using namespace std;
int a[50005], b[50005];
int dp[50005][2];
int main(){
int N;
cin >> N;
for(int i=1; i<=N; i++)
cin >> b[i];
int ans=0;
memset(dp, 0, sizeof(dp));
for(int i=2; i<=N; i++){
dp[i][0]=max(dp[i-1][0], dp[i-1][1]+abs(b[i-1]-1));//取1;
dp[i][1]=max(dp[i-1][0]+abs(b[i]-1), dp[i-1][1]+abs(b[i]-b[i-1]));//取峰值;
ans=max(dp[i][0], dp[i][1]);
}
cout << ans << endl;
return 0;
}