C. Remove the Bracket codeforces 1787C

Problem - C - Codeforces

题目大意：给出一长度为n的的数组，和一个数s，将第2到n-1个数分别分成两个数xi,yi，(xi-s)*(yi-s)0，求的最大值

3<=n<=2e5;0<=s<=2e5;0<=ai<=2e5

思路：因为数组中的数都是正整数，所以(xi-s)*(yi-s)0就等价于min(xi,yi)s或max(xi,yi)s。

要把一个数分成两个数，最优的方案就是一个是合法的最大数，另一个是最小的，因为只要这个数原来两边的数和不一样，如果分成的两个数一个数+1，一个数-1，那么乘积的和就会+或-和的差，所以只有xi,yi都是极端状态才可能成为最优方案。所以当这个数小于等于s时，xi,yi中的最大值小于等于s，所以就是这个数本身，最小值就是0，当这个数大于s时，如果这个数小于等于两倍的s，那么最大值就是s，最小值就是ai-s，如果这个数大于两倍的s，那么最大值是ai-s，最小值是s，所以综上，这个数大于s时最大值就是max(s,ai-s)，最小值等于ai-最大值。那么我们dp枚举即可，dp[i][0]表示当前数分成最大数在前，最小数在后，dp[i][1]表示当前数分成最大数在后，最小数在前，每个dp[i]都从dp[i-1]的两种状态中的最优值转移即可

//#include<__msvc_all_public_headers.hpp>
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 2e5 + 5;
ll ma[N], mi[N], dp[N][2];
int main()
{
	cin.tie(0);
	ios::sync_with_stdio(false);
	int t;
	cin >> t;
	while (t--)
	{
		ll n, s;
		cin >> n >> s;
		for (int i = 1; i <= n; i++)
		{
			ll x;
			cin >> x;
			if (i == 1)
			{//第一个数不能分
				ma[i] = mi[i] = x;
				continue;
			}
			if (i == n)
			{//最后一个数不能分
				ma[i] = mi[i] = x;
				continue;
			}
			if (x <= s)
			{//当前数小于等于s
				ma[i] = x;//最大的数就是这个数本身
				mi[i] = 0;//最小的数是0
			}
			else
			{
				ll x1 = s;
				ll x2 = x - s;
				ma[i] = max(x1, x2);//当前数大于s时最大值就是s和a[i]-s的最大值
				mi[i] = x - ma[i];//最小值是a[i]-ma[i]
			}
		}
		for (int i = 2; i <= n - 1; i++)
		{
			dp[i][0] = min(dp[i - 1][0] + mi[i - 1] * ma[i], dp[i - 1][1] + ma[i - 1] * ma[i]);//当前数分成最大数在前，分别从前一个数的两个状态中取最优转移过来
			dp[i][1] = min(dp[i - 1][0] + mi[i - 1] * mi[i], dp[i - 1][1] + ma[i - 1] * mi[i]);//前数分成最大数在后，分别从前一个数的两个状态中取最优转移过来
		}
		cout << min(dp[n - 1][0] + mi[n - 1] * ma[n], dp[n - 1][1] + ma[n - 1] * ma[n]) << endl;//最后一个数也要从前面转移过来
	}
	return 0;
}