51nod 1270 数组的最大代价 dp

题目链接：https://www.51nod.com/onlineJudge/questionCode.html#!problemId=1270

题目：

数组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

其实只要考虑Ai取1或者去Bi两种情况就行，然后一个DP就过了。
#include <iostream>
#include<bits/stdc++.h>

using namespace std;

int n,a=0,b,d[2][2];
int main()
{
cin>>n;
for(int i=0;i<n;i++)
{
scanf("%d",&b);
if(i)
{
d[0][1]=max(d[0][0],d[1][0]+a-1);
d[1][1]=max(d[0][0]+b-1,d[1][0]+abs(b-a));
}
d[0][0]=d[0][1];
d[1][0]=d[1][1];
a=b;
}
cout<<max(d[0][0],d[1][1])<<endl;
}