CF885B Marvolo Gaunt‘s Ring

声明

此题解借鉴了luogu大佬BoAn的题解此处仅为记录自己的做题过程

题目大意

给出一个有n个数的序列$a[1],a[2]\dots \dots a[n]$，使得在$i<=j<=k$的条件下，令$p\ast a[i]+q\ast a[j]+r\ast a[k]$的值最大并输出这个值

输入格式

第一行四个数$n、p、q、r$，第二行$n$个数，第$i$个数为$a[i]$的值

输出格式

一行，为$p\ast a[i]+q\ast a[j]+r\ast a[k]$的最大值$(i<=j<=k)$

思路

CF上也有标签显示这道题目可以用DP做
那我们考虑DP的做法
首先状态表示为
$$\begin{gathered} f_{i,1}表示前i个数中p\ast a_i的最大值 \\ {f}_{i,2}表示前i个数中p\ast a_i+q\ast a_j的最大值 \\ {f}_{i,3}表示前i个数中p\ast a_i+q\ast a_j+r\ast a_k的最大值 \\ p,q,r,a_i,a_j,a_k与题目表示的意思相同 \end{gathered}$$
那么状态转移方程为
$$\begin{gathered} f_{i,1}=max(f_{i-1,1},p\ast a_i) \\ {f}_{i,2}=max(f_{i-1,2},f_{i,1}+q\ast a_i) \\ {f}_{i,3}=max(f_{i-1,3},f_{i,2}+r\ast a_i) \end{gathered}$$
每次只需要便利一次数组时间复杂度为$O(n)$

同时提醒一下

1、考虑到有负数，如果不特判，最大值为$0$
2、我习惯性开long long的所以不知道会不会爆int，但是大佬的题解里提醒说不开long long可能会见祖宗

Code:

#include <iostream>

using namespace std;

#define int long long
const int N = 100010;

int f[N][4];
int n, p, q, r;
int a[N];

signed main()
{
	cin >> n >> p >> q >> r;
	for (int i = 1; i <= n; i++) cin >> a[i];
	f[1][1] = p * a[1];
	for (int i = 2; i <= n; i++) f[i][1] = max(f[i - 1][1], p * a[i]);
	f[1][2] = f[1][1] + q * a[1];
	for (int i = 2; i <= n; i++) f[i][2] = max(f[i - 1][2], f[i][1] + q * a[i]);
	f[1][3] = f[1][2] + r * a[1];
	for (int i = 2; i <= n; i++) f[i][3] = max(f[i - 1][3], f[i][2] + r * a[i]);
	cout << f[n][3];
	return 0;
}