题目大意:给出一长度为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;
}