【DP】Remove the Bracket—CF1787C

Remove the Bracket—CF1787C

翻译

RSJ有一个包含$n$个整数$a_1,a_2,\dots ,a_n$和一个整数$s$的序列$a$。对于$a_2,a_3,\dots ,a_{n-1}$中的每个元素，他选择了一对非负整数$x_i$和$y_i$，使得$x_i+y_i=a_i$且$(x_i-s)\cdot (y_i-s)\ge 0$。

现在他想要计算如下值：

$$F=a_1\cdot x_2+y_2\cdot x_3+y_3\cdot x_4+\dots +y_{n-2}\cdot x_{n-1}+y_{n-1}\cdot a_n.$$

请帮助他通过选择$x_i$和$y_i$的最佳值，找到可能的最小值$F$。可以证明至少存在一种有效的选择方法。

输入

每个测试包含多个测试用例。第一行包含一个整数$t$ ($1\le t\le 10^4$) —— 测试用例的数量。

每个测试用例的第一行包含两个整数$n$和$s$ ($3\le n\le 2\cdot 10^5$; $0\le s\le 2\cdot 10^5$)。

第二行包含$n$个整数$a_1,a_2,\dots ,a_n$ ($0\le a_i\le 2\cdot 10^5$)。

保证$n$的总和不超过$2\cdot 10^5$。

输出

对于每个测试用例，打印$F$的可能的最小值。

注意

在第一个测试用例中，$2\cdot 0+0\cdot 1+0\cdot 3+0\cdot 4=0$。

在第二个测试用例中，$5\cdot 1+2\cdot 2+2\cdot 2+1\cdot 5=18$。

思路

易知：$x_i>=s,y_i>=s$ 或 $x_i<=s,y_i<=s$。
由于需要得到最小的 $F$，所以 $F$ 右边的乘积式中的因子的差要尽可能大。
所以 $a_i$ 分成的 $x_i,y_i$ 一定满足其中一个是最大值，另外一个是最小值。
即对于每个 $i\in [2,n-1]$ 进行dp（$x_i$ 和 $y_i$ 中哪个取最大、哪个取最小）。

$Code$

#include <bits/stdc++.h>
#define int long long
#define sz(a) ((int)a.size())
using namespace std;
using PII = pair<int, int>;
using i128 = __int128;
const int N = 2e5 + 10;

int n, s;
int a[N];

void solve() {
	cin >> n >> s;
	for (int i = 1; i <= n; i ++) cin >> a[i];
	
	vector <array<int, 2>> f(n + 1, {0, 0});
	// dp数组，f[i][0]：x_i取最小，y_i取最大
	vector <int> imin(n + 1, 0), imax(n + 1, 0);
	
	for (int i = 1; i <= n; i ++) {
		if (a[i] >= 2 * s) { // x_i、y_i都 >= s
			imin[i] = s;
			imax[i] = a[i] - s;
		} else {  // x_i、y_i都 <= s
			imax[i] = min(a[i], s);
			imin[i] = a[i] - imax[i];
		}
	}
	
	f[2][0] = a[1] * imin[2];
	f[2][1] = a[1] * imax[2];
	for (int i = 3; i <= n - 1; i ++) {
		f[i][0] = min(f[i - 1][0] + imax[i - 1] * imin[i], f[i - 1][1] + imin[i - 1] * imin[i]);
		f[i][1] = min(f[i - 1][0] + imax[i - 1] * imax[i], f[i - 1][1] + imin[i - 1] * imax[i]);
	}
	
	cout << "       ";
	cout << min(f[n - 1][0] + imax[n - 1] * a[n], f[n - 1][1] + imin[n - 1] * a[n]) << "\n";
	// 注意：最后的y_n-1 + a[n]不能在上边那个for循环中进行，
	// 因为n <= 3的时候是无法进入循环的！
}

signed main() {
	ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
	int t = 1;
	cin >> t; cin.get();
	while (t --) solve();
	return 0;
}