【ybtoj高效进阶 21173】简单区间（分治）

简单区间

题目链接：ybtoj高效进阶 21173

题目大意

给你一个数组，问你有多少个区间，使得它们的和减去它们的最大值是 k 的倍数。

思路

我们考虑进行分治。

考虑两个指针扫。
两边轮流进行，当确定一个作为最大值的时候，你就把当前那一部分的和减去最大值，然后另外一边那个表记录一下和取模之后是某个值的个数，然后直接加就好了。

当然你用笛卡尔树暴力合并加个启发式合并也可以过，但就是 \(O(nlog^2n)\) 的了。

代码

#include<cstdio>
#include<iostream>
#define ll long long

using namespace std;

int n, k, maxn[300001], dy[300001], a[300001];
int s[300001], nm1[1000001], nm[1000001];
ll ans;

void work(int l, int r) {//分治
	if (l == r) return ;
	int mid = (l + r) >> 1;
	work(l, mid); work(mid + 1, r);
	
	maxn[0] = 0;
	s[mid] = 0; int sum = 0, maxx = 0;
	for (int i = mid + 1; i <= r; i++) {
		if (a[i] > a[maxn[maxn[0]]]) maxn[++maxn[0]] = i;
		s[i] = (s[i - 1] + a[i]) % k;
		nm[((s[i] - a[maxn[maxn[0]]]) % k + k) % k]++;
		dy[i] = maxn[maxn[0]];
	}
	maxn[maxn[0] + 1] = r + 1;
	int l1 = 1, l2 = mid + 1;
	for (int i = mid; i >= l; i--) {
		sum = (sum + a[i]) % k;
		maxx = max(maxx, a[i]);
		while (l1 <= maxn[0] && a[maxn[l1]] <= maxx) l1++;
		while (l2 < maxn[l1]) {
			nm[((s[l2] - a[dy[l2]]) % k + k) % k]--;
			nm1[s[l2]]++;
			l2++;
		}
		if (l1 <= maxn[0]) ans += nm[(k - sum) % k];
		ans += nm1[(maxx % k - sum + k) % k];
	}
	for (int i = mid + 1; i < l2; i++) nm1[s[i]]--;//清空
	for (int i = l2; i <= r; i++) nm[((s[i] - a[dy[i]]) % k + k) % k]--;
}

int main() {
//	freopen("interval.in", "r", stdin);
//	freopen("interval.out", "w", stdout);
	
	scanf("%d %d", &n, &k);
	for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
	
	work(1, n);
	printf("%lld", ans);
	
	return 0;
}