BZOJ_3343_教主的魔法_分块+二分查找
题意:教主最近学会了一种神奇的魔法,能够使人长高。于是他准备演示给XMYZ信息组每个英雄看。于是N个英雄们又一次聚集在了一起,这次他们排成了一列被编号为1、2、……、N。每个人的身高一开始都是不超过1000的正整数。教主的魔法每次可以把闭区间[L, R](1≤L≤R≤N)内的英雄的身高全部加上一个整数W。(虽然L=R时并不符合区间的书写规范,但我们可以认为是单独增加第L(R)个英雄的身高)询问WD闭区间 [L, R] 内有多少英雄身高大于等于C,以验证教主的魔法是否真的有效。
分析:分块。
每一块用一个辅助数组来排序,询问时整块二分查找大于等于C的位置,零散的暴力重构。
重构时可以归并也可以直接sort
代码:
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <math.h>
using namespace std;
#define N 1000010
#define LL long long
void read(LL &x){
int f=1;x=0;char s=getchar();
while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
while(s>='0'&&s<='9'){x=(x<<3)+(x<<1)+s-'0';s=getchar();}
x*=f;
}
int L[1010],R[1010],pos[N],t,is[1010],add[1010],n,q;
LL a[N],b[N];
int find(int blo,LL x){
int l=L[blo],r=R[blo]+1;
while(l<r){
int mid=l+r>>1;
if(b[mid]<x)l=mid+1;
else r=mid;
}return R[blo]-l+1;
}
inline void pre(int blo){
for(int i=L[blo];i<=R[blo];i++)b[i]=a[i];
sort(b+L[blo],b+R[blo]+1);
}
void up(int l,int r,LL c){
int p=pos[l],q=pos[r];
if(p==q){
for(int i=l;i<=r;i++){
a[i]+=c;
}
pre(p);
}else{
for(int i=p+1;i<q;i++){
add[i]+=c;
}
for(int i=l;i<=R[p];i++){
a[i]+=c;
}
for(int i=L[q];i<=r;i++){
a[i]+=c;
}
pre(p);pre(q);
}
}
int query(int l,int r,LL c){
int p=pos[l],q=pos[r];
int ans=0;
if(p==q){
for(int i=l;i<=r;i++)if(a[i]+add[p]>=c)ans++;
}else{
for(int i=p+1;i<q;i++){
ans+=find(i,c-add[i]);
}
for(int i=l;i<=R[p];i++){
if(a[i]+add[p]>=c)ans++;
}
for(int i=L[q];i<=r;i++){
if(a[i]+add[q]>=c)ans++;
}
}
return ans;
}
char op[10];
int main(){
scanf("%d%d",&n,&q);
t=sqrt(n);
for(int i=1;i<=t;i++){
L[i]=R[i-1]+1,R[i]=t*i;
for(int j=L[i];j<=R[i];j++){
read(a[j]);pos[j]=i;
}
}
if(n-t*t){
t++;L[t]=R[t-1]+1;R[t]=n;
for(int i=L[t];i<=R[t];i++){
read(a[i]);pos[i]=t;
}
}
for(int i=1;i<=t;i++)pre(i);
int x,y,z;
while(q--){
scanf("%s%d%d%d",op,&x,&y,&z);
if(op[0]=='M'){
up(x,y,1ll*z);
}else{
printf("%d\n",query(x,y,1ll*z));
}
}
}