GDGYOJ 马子哥的篮球梦
算法:动态规划
-
状态定义: f ( i , j ) f(i, j) f(i,j) 表示如下 —— f ( i , 0 ) f(i, 0) f(i,0) 表示在前 i i i 个数中,不选第 i i i 个数作为子序列的结尾的方案; f ( i , 1 ) f(i, 1) f(i,1) 表示在前 i i i 个数中,选择第 i i i 个数作为子序列的结尾且 i i i 是选择的第奇数个,即选择的子序列长度为奇数; f ( i , 2 ) f(i, 2) f(i,2) 表示在前 i i i 个数中,选择第 i i i 个数作为子序列的结尾且 i i i 是选择的第偶数个,即选择的子序列长度为偶数。
-
状态转移: f ( i , 0 ) = m a x ( f ( j , 1 ) , f ( k , 2 ) ) f(i, 0) = max(f(j, 1), f(k, 2)) f(i,0)=max(f(j,1),f(k,2)),其中 j j j 和 k k k 分别为前 i − 1 i - 1 i−1 个数中长度是奇数和偶数的最大”手感“和对应的序列最右端的下标; f ( i , 1 ) = f ( k , 2 ) + a [ i ] f(i, 1) = f(k, 2) + a[i] f(i,1)=f(k,2)+a[i],表示以 i i i 结尾的奇数长子序列需要接在前 i − 1 i - 1 i−1 个数中以 k k k 结尾的偶数长子序列的后面,而奇数需要加上当前手感,偶数需要减去,因此另一个转移方程为 f ( i , 2 ) = f ( j , 1 ) − a [ i ] f(i, 2) = f(j, 1) - a[i] f(i,2)=f(j,1)−a[i]。
-
一开始都置为非法状态,由于是求最大值,因此都设为负无穷。
-
边界条件: f ( 0 , 0 ) = 0 f(0, 0) = 0 f(0,0)=0 表示以 0 0 0 结尾的子序列,不选当前 0 0 0 的方案为 0 0 0; f ( 0 , 2 ) = 0 f(0, 2) = 0 f(0,2)=0 表示以 0 0 0 结尾的偶数长子序列,同样合法。而 f ( 0 , 1 ) f(0, 1) f(0,1) 表示的是以 0 0 0 结尾的奇数长子序列,一定不合法,因为 0 0 0 是不存在的数,只是一个边界,只能作为第一个被选中进入子序列的数的前驱,所以只能为偶数或者不选。
时间复杂度 O ( n ) O(n) O(n)
C ++ 代码
#include <iostream>
#include <algorithm>
#include <cstring>
#define endl '\n'
using namespace std;
typedef long long LL;
const int N = 1000010;
int n;
int a[N];
LL f[N][3];
int main()
{
cin >> n;
for (int i = 1; i <= n; i ++ ) cin >> a[i];
memset(f, -0x3f, sizeof f);
f[0][0] = f[0][2] = 0;
for (int i = 1, j = 0, k = 0; i <= n; i ++ )
{
f[i][0] = max(f[j][1], f[k][2]);
f[i][1] = f[k][2] + a[i];
f[i][2] = f[j][1] - a[i];
if (f[j][1] < f[i][1]) j = i;
if (f[k][2] < f[i][2]) k = i;
}
cout << max({0ll, f[n][0], f[n][1], f[n][2]}) << endl;
return 0;
}