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
1 2 3 4 5
A 1 5 4
M 3 5 1
A 1 5 4
Sample Output
2
3
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(懒得算了反正能过)
先分块,用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 }