【每日一题】codeforces 1358D (尺取法)

每日一题，坚持使我强大

---

今日份快乐：codeforces 1358D 传送门

明天份快乐：codeforces 459C 传送门

---

题目大意

在一个神奇的地方，这里有 n 个月份，第 i 个月份有 a[i] 天。 在这里每个月份的第一天会得到 1 点价值，第二天会得到 2 点价值……依次类推到这个月份的最后一天。
问：有个人在要这里连着待 x 天，最多可以获得多少价值。

分析

这个题用尺取法就很合适，我们一步一步分析。

---

如果一个月有 60 天，在这个月份里待 5 天的话，我们肯定是选最后五天 {56，57，58，59，60}，其他方案肯定没有这五天得到的价值多。
推广一下：有 n 个月，天数最多的月份为 60 天，如果再待 5 天的话，我们肯定还是选这个月的最后五天 {56，57，58，59，60}。如果我们在这里待 61 天呢？显然的，我们要选这个月的 60 天，剩下的一天我们肯定是选前一个月的最后一天，而不是后一个月的第一天。
如果不是太清楚就看下边这个表：

1	2	3	4	1	2	3	4	5	6
2	3	4	1	2	3	4	5	6	7

假设只有两个月份，第一个月有 4 天， 第二个月有 7 天，在这里待 10 天的话，两种选择的差别一眼就可以看出，要选有最后一个月的最后一天的方案。所以，无论有几个月，每个月有几天，最终的方案数肯定是从某天到往后的某个月的最后一天。

---

接下来，我们先讲一下尺取法：
尺取法是一种高效的枚举，拿这个题来说，我们先从第一个月份开始，找个一个满足的题意的方案，这个方案就是一个区间，我们定义为左端为 L，右端为 R。L 到 R 的区间内为满足题意的方案，更新答案。接下来我们先让 R 右移，这时肯定不再满足题意，为了使我们选取的区间满足题意，我们再让 L 右移，使这个区间满足题意，再次更新答案，依次类推，直到最后一个方案（或者说类推到数组的边界）。如下图：

我们令 x = 18(方格内为这个月的天数)，这时的 L 和 R 选取的区间正好为一个满足题意区间，知道每个月的天数 a[i] 肯定可以计算出每个月的价值 sum，sum = (a[i] + 1) * a[i] / 2 (等差数列)。我们计算出所有月份的的价值统计一下去更新答案就ok。接下来我让 R 右移：

这时，区间内的总天数为 21，为了合题意，我们让 L 右移：

这时区间内的 x 正好为 18，再次更新答案，然后右移 R，同时，为了满足题意我们也右移依次 L：

这时候，我们可以发现，现在 x 的值为 20，不满足题意，为了满足题意，我们要去掉两天，根据我们上边的结论，我们去掉 L 指向月份的前两天，在这个月的总价值中减去 3。如果我们要去掉 t 天，应该减去的价值为 t * (t + 1) / 2。我们依次类推遍历整个数组就OK。

尺取法就是通过维护 L 和 R，使其满足题意，进而遍历到所有的方案，是一种高效的枚举，它比暴力枚举要高效很多，但没具体固定的写法，不同的题目，维护的方法不同。但核心思想不变：L 和 R 之间是一组满足题意的方案。

---

这个题还有一个坑点，今年的后几个月到明年的前几个月也可能凑成答案，解决方法在代码中

代码

#include <bits/stdc++.h>
using namespace std;

#define ll long long
const int maxn = 4e5+5;
ll a[maxn] = {0}; 

int main() {

	ios::sync_with_stdio(false);
	
	ll n, x;
	cin >> n >> x;
	for(int i = 1; i <= n; i++) cin >> a[i];
	
	ll res = 0, sum = 0, num = 0;		// sum 纪录 L 到 R 的总价值和，num 为总天数，res 为最终的结果
	int l = 0, r = 1;			// 我们让 L 从第 0 个月(假设为 0 天，不会影响答案) 开始到第 n 个月，这就可以遍历所有方案
	while(l <= n){
		
		if(r > n) r -= n;		// 这里就是解决跨年的方案的办法，当 r 超过 n 时，使 r 减去 n 得到的便是下一年 r 在的月份
		sum += a[r] * (a[r] + 1) / 2;	// r 右移过后要先加上新指向的月份的价值
		num += a[r];			// r 右移过后要先加上新指向的月份的天数
		
		if(num >= x){			// 如果这时区间内的天数超过 x，就右移 L
			while(num - a[l] > x && l <= n){	// L 右移
				num -= a[l];			// 更新天数
				sum -= a[l] * (a[l] + 1) / 2;	// 更新价值
				l++;				
			}
			ll tmp = num - x;			// 减去多出的天数
			res = max(res, sum - tmp *(tmp + 1) / 2); // 减去多出的价值，并更新答案
		}
		
		r++;
	}
	
	cout << res << endl;		
	
    return 0;
}

---

写在最后

一定要注意数据范围，不然就会这样…

欢迎大佬留言指错，哪里讲的不清楚也欢迎留言提出