Educational Codeforces Round 63 (Rated for Div. 2) D. Beautiful Array（DP：记录状态）

题目链接：

D. Beautiful Array

 

题意：

给定一个序列和x，可以任选一段连续区间，将其中所有元素*x，（也可以不选），求序列的最大连续子段和。

 

思路：

设 dp[i][j][k] ：

i：表示当前位置

j：0：还未开始计算最大连续子段和

    1：正在计算最大连续子段和

    2：计算最大连续子段和已经结束

k：0：还未开始*x

     1：正在*x

     2：*x已结束

状态转移：

在同一位置：

dp[i][j][k] -> dp[i][j+1][k] ；

dp[i][j][k] -> dp[i][j][k+1] ；

在不同位置：

dp[i][j][k] = max(dp[i][j][k]，dp[i-1][j][k] + (j==1?a[i]:0)*(k==1?x:1)) ；

 

Code：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

const int MAX = 3e5 + 100;
const ll inf = 1e18;

ll n, x;
ll a[MAX];
ll dp[MAX][5][5];

int main()
{
	scanf("%lld%lld", &n, &x);
	for (int i = 1; i <= n; i++) {
		scanf("%lld", &a[i]);
	}
	for (int i = 0; i <= n; i++) {
		for (int j = 0; j < 3; j++) {
			for (int k = 0; k < 3; k++) {
				dp[i][j][k] = -inf;
			}
		}
	}
	dp[0][0][0] = 0;
	//注意 i = 0 的时候需要在同一位置进行转移
	for (int i = 0; i <= n; i++) {
		for (int j = 0; j < 3; j++) {
			for (int k = 0; k < 3; k++) {
				if (j >= 1)
					dp[i][j][k] = max(dp[i][j][k], dp[i][j - 1][k]);
				if (k >= 1)
					dp[i][j][k] = max(dp[i][j][k], dp[i][j][k - 1]);
				if (i >= 1)
					dp[i][j][k] = max(dp[i][j][k], dp[i - 1][j][k] + 1ll * (j == 1 ? a[i] : 0) * (k == 1 ? x : 1));
			}
		}
	}
	printf("%lld\n", dp[n][2][2]);
	return 0;
}

/*
6 0
4 -1 -1 -1 6 -4
4 -1
-1000 100 -1 -100
*/