思路:贪心一下,要使拥抱次数最多必然是以每一个月的末尾为最后一天时,假设b(n-3) b(n-2) b(n-1) b(n) …c(1) c(2)… c(n)…a(n-4) a(n-3) 完后挪一个 b(n-2) b(n-1) b(n) …a(n-3) a(n-2) 当a(n-2)>b(n-3)时会使值增加,那么肯定继续挪到最后一个时a(n)最优。而a(n-2)<b(n-3)时我们往回挪使值增大,也挪到c(n)时最优。
现在可以知道我们只需枚举每一个最后的点即可,再前缀和记录预处理一下,求每个截至时贪心比较即可得出答案,细节见代码。
Code:
#include<iostream>
#include<string>
#include<map>
#include<algorithm>
#include<memory.h>
#include<cmath>
#define pii pair<int,int>
#define FAST ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
using namespace std;
typedef long long ll;
const int Max = 2e6 + 5;
int lst[Max];
ll sx[Max], s[Max];
int main()
{
ll n, x;cin >> n >> x;
for (int i = 1;i <= n;i++)
{
cin >> sx[i];sx[i + n] = sx[i];
s[i+n]=s[i] = (sx[i] + 1) * sx[i] / 2;
}
for (int i = 1;i <= 2*n;i++)
{
sx[i] = sx[i-1] + sx[i];s[i] = s[i-1] + s[i];
}
ll ans = 0;
for (int i = 1;i <= n * 2;i++)
{
if (sx[i] < x)continue;
ll low = sx[i] - x;
ll p = lower_bound(sx + 1, sx + 2 * n + 1, low) - sx;
if (x == sx[i] - sx[p])ans = max(ans, s[i] - s[p]);
else
{
ll dnum = low - sx[p - 1];
ans = max(ans, s[i] - s[p - 1] - (dnum + 1)*dnum / 2);
}
}
cout << ans << endl;
}