3343: 教主的魔法
Time Limit: 10 Sec Memory Limit: 256 MB
Submit: 614 Solved: 265
[Submit][Status][Discuss]
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。
Source
一道十分裸的分块题。。。
记得讲分块的时候已经是几个月前的事情了。。然而我现在才开始填这个坑。。
这道分块很经典。。。区间加,求区间中大于一个值的数数量是多少
我们把序列分成 \(\sqrt n\) 块,对于每一块我们对块内的数从大到小排序,查询的时候对中间完整的块二分查找符合条件的数,两边的块暴力做。修改也一样。最终复杂度是\(O(n*\sqrt n)\)
#include <cmath> #include <cstdio> #include <cstring> #include <algorithm> using namespace std; const int N = 10000010; int b[N],a[N],pos[N],add[N],n,Q,m,blk,l,r,val; char opt[5]; int read(){ int x = 0,f = 1,c = getchar(); while (c<'0' || c>'9') {if (c=='-') f = -1;c = getchar();} while (c>='0' && c<='9') x = x*10+c-'0',c = getchar(); return x*f; } bool cmp(int a,int b){return a>b;} void reset(int x){ int l = (x-1)*blk+1,r = min(n,x*blk); for (int i=l;i<=r;i++) b[i] = a[i]; sort(b+l,b+r+1,cmp); } int find(int x,int val){ int l = blk*(x-1)+1,r = blk*x,ans = 0; while (l<=r){ int mid = (l+r)>>1; if (b[mid]>=val){ans = mid-blk*(x-1);l = mid+1;} else r = mid-1; } return ans; } void Update(int l,int r,int val){ if (pos[l]==pos[r]) for (int i=l;i<=r;i++) a[i]+=val; else { for (int i=l;i<=min(n,blk*pos[l]);i++) a[i]+=val; for (int i=(pos[r]-1)*blk+1;i<=r;i++) a[i]+=val; } reset(pos[l]);reset(pos[r]); for (int i=pos[l]+1;i<pos[r];i++) add[i]+=val; } int Query(int l,int r,int val){ int res = 0; if (pos[l]==pos[r]) for (int i=l;i<=r;i++) {if (a[i]>=val-add[pos[i]]) res++;} else { for (int i=l;i<=min(n,blk*pos[l]);i++) if (a[i]>=val-add[pos[i]]) res++; for (int i=(pos[r]-1)*blk+1;i<=r;i++) if (a[i]>=val-add[pos[i]]) res++; } for (int i=pos[l]+1;i<pos[r];i++) res+=find(i,val-add[i]); return res; } int main(){ n = read();Q = read(); blk = (int)sqrt(n); for (int i=1;i<=n;i++){ a[i] = read(); pos[i] = (i-1)/blk+1; b[i] = a[i]; } if (n%blk) m = n/blk+1; else m = n/blk; for (int i=1;i<=m;i++) reset(i); for (int i=1;i<=Q;i++){ scanf("%s",opt);l = read();r = read();val = read(); if (opt[0]=='M') Update(l,r,val); else printf("%d\n",Query(l,r,val)); } }