题意:
一种新的纪年方式,一年有 个月,每个月有 天,没有闰年。计算某一天的值的方式为该天在该月中是第几天。连续选择 天,要求这 天得到的值最大。
最优解的区间的右端点一定一定是某个月的月末,枚举每个月的月末,二分查找区间左端点即可。
AC代码:
const int N = 1e6 + 50;
ll n, m, x;
ll a[N], s[N], sum[N];
int main()
{
int t;
sldd(n, x);
s[0] = 0;
sum[0] = 0;
rep(i, 1, n)
{
sld(a[i]);
s[i] = s[i - 1] + a[i];
sum[i] = sum[i - 1] + (1 + a[i]) * a[i] / 2; //等差数列和
}
rep(i, n + 1, 2 * n)
{
a[i] = a[i - n];
s[i] = s[i - 1] + a[i];
sum[i] = sum[i - 1] + (1 + a[i]) * a[i] / 2;
}
n *= 2;
ll ans = 0;
rep(i, 0, n)
{
ll p = upper_bound(s, s + n + 1, s[i] - x) - s;
if (p == 0)
continue;
ll t = s[i] - x - s[p - 1];
ans = max(ans, sum[i] - sum[p - 1] - (1 + t) * t / 2);
}
pld(ans);
return 0;
}