题目链接点这里
题目描述
教主最近学会了一种神奇的魔法,能够使人长高。于是他准备演示给XMYZ信息组每个英雄看。于是N个英雄们又一次聚集在了一起,这次他们排成了一列,被编号为1、2、……、N。
每个人的身高一开始都是不超过1000的正整数。教主的魔法每次可以把闭区间[L, R](1≤L≤R≤N)内的英雄的身高全部加上一个整数W。(虽然L=R时并不符合区间的书写规范,但我们可以认为是单独增加第L(R)个英雄的身高)
CYZ、光哥和ZJQ等人不信教主的邪,于是他们有时候会问WD闭区间 [L, R] 内有多少英雄身高大于等于C,以验证教主的魔法是否真的有效。
WD巨懒,于是他把这个回答的任务交给了你。
输入格式
第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。
输出格式
对每个“A”询问输出一行,仅含一个整数,表示闭区间 [L, R] 内身高大于等于C的英雄数。
输入输出样例
输入
5 3
1 2 3 4 5
A 1 5 4
M 3 5 1
A 1 5 4
输出
2
3
说明/提示
【输入输出样例说明】
原先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。解析:
分块,是一种优雅的暴力,复杂度为O(sqrt(n)),可以处理区间类的一些问题,下面结合代码体会一下分块儿的思想.
我是看了 Lewis Kulen大佬总结的博客,才感触颇多。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef pair<int,int> P;
#define faster ios::sync_with_stdio(0),cin.tie(0),cout.tie(0)
const int maxn=1e6+10;
const int inf=0x3f3f3f3f;
int n,q;
int a[maxn];//原数组
int b[maxn];//分块后排完序的数组
int L[maxn],R[maxn];//每块的左边界,右边界
int belong[maxn];//各元素所属的块的编号
int lazy[maxn/100];//各块的lazy标记,可以结合线段树的lazy理解
int num,block;//块的数量,块的大小
void build(){//初始化
block=(int)(sqrt(n*1.0));
num=n/block;
if(n%block)++num;//最后一块可能是不完整的
for(int i=1;i<num;i++){
L[i]=(i-1)*block+1;
R[i]=i*block;
sort(b+L[i],b+R[i]+1);//块内保证有序,便于二分查找想要的数
}
L[num]=(num-1)*block+1;R[num]=n;//最后一块特殊处理
sort(b+L[num],b+n+1);
for(int i=1;i<=n;i++){//第i个数所属块为belong[i]
belong[i]=(i-1)/block+1;
}
}
void reset(int x){//保证处理后块内依然有序
for(int i=L[x];i<=R[x];i++){
b[i]=a[i];
}
sort(b+L[x],b+R[x]+1);
}
void update(int l,int r,int v){//区间加操作
if(belong[l]==belong[r]){
for(int i=l;i<=r;i++)a[i]+=v;
reset(belong[l]);
return ;
}
for(int i=l;i<=R[belong[l]];i++)a[i]+=v;
reset(belong[l]);
for(int i=belong[l]+1;i<belong[r];i++)lazy[i]+=v;
for(int i=L[belong[r]];i<=r;i++)a[i]+=v;
reset(belong[r]);
}
int fin(int x,int v){//二分查找块内符合要求的数的个数
int now=lower_bound(b+L[x],b+R[x]+1,v)-b-L[x];
return block-now;
}
int query(int l,int r,int c){
int sum=0;
if(belong[l]==belong[r]){
for(int i=l;i<=r;i++){
if(a[i]+lazy[belong[l]]>=c)sum++;
}
return sum;
}
for(int i=l;i<=R[belong[l]];i++){
if(a[i]+lazy[belong[l]]>=c)sum++;
}
for(int i=belong[l]+1;i<belong[r];i++){
sum+=fin(i,c-lazy[i]);
}
for(int i=L[belong[r]];i<=r;i++){
if(a[i]+lazy[belong[r]]>=c)sum++;
}
return sum;
}
int main()
{
faster;
cin>>n>>q;
for(int i=1;i<=n;i++)cin>>a[i],b[i]=a[i];
build();
char s;int l,r,x;
for(int i=1;i<=q;i++){
cin>>s>>l>>r>>x;
if(s=='M'){
update(l,r,x);
}
else {
cout<<query(l,r,x)<<endl;
}
}
return 0;
}