3343. 教主的魔法【分块】

Description

教主最近学会了一种神奇的魔法，能够使人长高。于是他准备演示给XMYZ信息组每个英雄看。于是N个英雄们又一次聚集在了一起，这次他们排成了一列，被编号为1、2、……、N。

每个人的身高一开始都是不超过1000的正整数。教主的魔法每次可以把闭区间[L, R]（1≤L≤R≤N）内的英雄的身高全部加上一个整数W。（虽然L=R时并不符合区间的书写规范，但我们可以认为是单独增加第L（R）个英雄的身高）

CYZ、光哥和ZJQ等人不信教主的邪，于是他们有时候会问WD闭区间 [L, R] 内有多少英雄身高大于等于C，以验证教主的魔法是否真的有效。

WD巨懒，于是他把这个回答的任务交给了你。

Input

第1行为两个整数N、Q。Q为问题数与教主的施法数总和。

第2行有N个正整数，第i个数代表第i个英雄的身高。

第3到第Q+2行每行有一个操作：

（1）若第一个字母为“M”，则紧接着有三个数字L、R、W。表示对闭区间 [L, R] 内所有英雄的身高加上W。

（2）若第一个字母为“A”，则紧接着有三个数字L、R、C。询问闭区间 [L, R] 内有多少英雄的身高大于等于C。

Output

对每个“A”询问输出一行，仅含一个整数，表示闭区间 [L, R] 内身高大于等于C的英雄数。

Sample Input

5 3
1 2 3 4 5
A 1 5 4
M 3 5 1
A 1 5 4

Sample Output

2
3

HINT

【输入输出样例说明】

原先5个英雄身高为1、2、3、4、5，此时[1, 5]间有2个英雄的身高大于等于4。教主施法后变为1、2、4、5、6，此时[1, 5]间有3个英雄的身高大于等于4。

【数据范围】

对30%的数据，N≤1000，Q≤1000。

对100%的数据，N≤1000000，Q≤3000，1≤W≤1000，1≤C≤1,000,000,000。

第一个写的分块题……挺有意思的
先分块，用a数组存原数，b数组存每个块内排序好的数
修改的时候，整块直接打标记，小块的话就在a数组内暴力修改，并将暴力修改的地方更新到b数组中并排好序
查询的时候，对两端暴力统计。中间的由于已经排序完，所以可以二分
复杂度O(懒得算了反正能过)

 1 #include<iostream>
 2 #include<cstring>
 3 #include<cstdio>
 4 #include<cmath>
 5 #include<algorithm>
 6 #define N (1000000+100) 
 7 using namespace std;
 8 int Add[N],unit,num;
 9 int n,m,a[N],b[N];
10 int L[N],R[N],ID[N];
11 void Build()
12 {
13     unit=sqrt(n);
14     num=n/unit;
15     if (n%unit) num++;
16     for (int i=1;i<=num;++i)
17         L[i]=(i-1)*unit+1,R[i]=i*unit;
18     R[num]=n;
19     for (int i=1;i<=n;++i)
20         ID[i]=(i-1)/unit+1;
21     for (int i=1;i<=num;++i)
22         sort(b+L[i],b+R[i]+1);
23 }
24 
25 void Resort(int x)
26 {
27     for (int i=L[x];i<=R[x];++i) b[i]=a[i];
28     sort(b+L[x],b+R[x]+1);
29 }
30 
31 void Update(int l,int r,int k)
32 {
33     if (ID[l]==ID[r])
34     {
35         for (int i=l;i<=r;++i)  a[i]+=k;
36         Resort(ID[l]);
37         return;
38     }
39     for (int i=l;i<=R[ID[l]];++i)  a[i]+=k;
40     for (int i=L[ID[r]];i<=r;++i)  a[i]+=k;
41     Resort(ID[l]);Resort(ID[r]);
42     for (int i=ID[l]+1;i<=ID[r]-1;++i)    Add[i]+=k;
43 }
44 
45 int Query(int l,int r,int k)
46 {
47     int ans=0;
48     if (ID[l]==ID[r])
49     {
50         for (int i=l;i<=r;++i) if (a[i]+Add[ID[i]]>=k) ans++;
51         return ans;
52     }
53     for (int i=l;i<=R[ID[l]];++i) if (a[i]+Add[ID[i]]>=k) ans++;
54     for (int i=L[ID[r]];i<=r;++i) if (a[i]+Add[ID[i]]>=k) ans++;
55     for (int i=ID[l]+1;i<=ID[r]-1;++i)
56     {
57         int p=lower_bound(b+L[i],b+R[i]+1,k-Add[i])-b;
58         ans+=R[i]-p+1;
59     }
60     return ans;
61 }
62 
63 int main()
64 {
65     int l,r,k;
66     char p;
67     scanf("%d%d",&n,&m);
68     for (int i=1;i<=n;++i)
69         scanf("%d",&a[i]),b[i]=a[i];
70     Build();
71     for (int i=1;i<=m;++i)
72     {
73         scanf("
%c%d%d%d",&p,&l,&r,&k);
74         if (p=='M')    Update(l,r,k);
75         if (p=='A')    printf("%d
",Query(l,r,k));
76     }
77 }