Codeforces 1358 D. The Best Vacation(二分)

在这里插入图片描述
在这里插入图片描述

题意:

一种新的纪年方式,一年有 n n 个月,每个月有 d i d_i 天,没有闰年。计算某一天的值的方式为该天在该月中是第几天。连续选择 x x 天,要求这 x x 天得到的值最大。

最优解的区间的右端点一定一定是某个月的月末,枚举每个月的月末,二分查找区间左端点即可。

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;
}

猜你喜欢

转载自blog.csdn.net/qq_43627087/article/details/106391202