Luogu P2801 教主的魔法

题目传送门

第一次分块

---

分块，将每一个块内部排序
对于A操作，块内部二分查找，两边的零散部分在原数组中暴力查找
对于M操作，给整块打上标记，两边的零散部分暴力修改

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#define LL long long
using namespace std;
LL read() {
	LL k = 0, f = 1; char c = getchar();
	while(c < '0' || c > '9') {
		if(c == '-') f = -1;
		c = getchar();
	}
	while(c >= '0' && c <= '9')
		k = k * 10 + c - 48, c = getchar();
	return k * f;
}
char read_c() {
	char c = getchar();
	while(c != 'M' && c != 'A') c = getchar();
	return c;
}
LL a[1000010], b[1000010], tag[1000010];
int l[1000010], r[1000010], cnt, belong[1000010];
int main() {
	//freopen("1.in", "r", stdin);
	int n = read(), q = read(), bl = sqrt(n); cnt = (n-1) / bl + 1;
	for(int i = 1; i <= n; ++i) {
		a[i] = b[i] = read(); int x = (i - 1) / bl + 1;
		belong[i] = x;
		if(belong[i] != belong[i-1]) l[x] = i, r[x-1] = i-1;
	}
	l[1] = 1, r[cnt] = n;
	//cout << cnt << endl;
	for(int i = 1; i <= cnt; ++i) sort(a+l[i], a+r[i]+1);
	while(q--) {
		//cout << q << endl;
		char opt = read_c();
		if(opt == 'A') {
			int nl = read(), nr = read(); LL k = read();
			int pl = belong[nl], pr = belong[nr];
			int ans = 0;
			if(pl == pr) {
				for(int i = nl; i <= nr; ++i) ans += (b[i] >= k-tag[pl]);
				printf("%d\n", ans); continue;
			}
			for(int i = pl+1; i <= pr-1; ++i)
				ans += (r[i]-l[i]+1) - (lower_bound(a+l[i], a+r[i]+1, k-tag[i]) - a - l[i]);
				
			if(nl == l[pl]) ans += (r[pl]-l[pl]+1) - (lower_bound(a+l[pl], a+r[pl]+1, k-tag[pl]) - a - l[pl]);
			else for(int i = nl; i <= r[pl]; ++i) ans += (b[i] >= k-tag[pl]);
			
			if(nr == r[pr]) ans += (r[pr]-l[pr]+1) - (lower_bound(a+l[pr], a+r[pr]+1, k-tag[pr]) - a - l[pr]);
			else for(int i = l[pr]; i <= nr; ++i) ans += (b[i] >= k-tag[pr]);
			printf("%d\n", ans);
		}
		else {
			int nl = read(), nr = read(); LL k = read();
			int pl = belong[nl], pr = belong[nr];
			if(pl == pr) {
				for(int i = nl; i <= nr; ++i) b[i] += k;
				for(int i = l[pl]; i <= r[pl]; ++i) a[i] = b[i];
				sort(a+l[pl], a+r[pr]+1);
			}
			for(int i = pl+1; i <= pr-1; ++i)
				tag[i] += k;
			if(nl == l[pl]) tag[pl] += k;
			else {
				for(int i = nl; i <= r[pl]; ++i) b[i] += k;
				for(int i = l[pl]; i <= r[pl]; ++i) a[i] = b[i];
				sort(a+l[pl], a+r[pl]+1);
			}
			if(nr == r[pr]) tag[pr] += k;
			else {
				for(int i = l[pr]; i <= nr; ++i) b[i] += k;
				for(int i = l[pr]; i <= r[pr]; ++i) a[i] = b[i];
				sort(a+l[pr], a+r[pr]+1);
			}
		}
	}
	
	return 0;
}