BZOJ[3343]教主的魔法 分块+二分

传送门ber~

分块题

构造：每个块排遍序，块大小为$\sqrt n$
加值：给完整的块打标记，不完整的块暴力重构(重新排序)
查询：在每个完整的块中二分查找，不完整的快中暴力判断

代码如下：

#include<algorithm>
#include<ctype.h>
#include<cstdio>
#include<cmath>
#define N 1000050
using namespace std;
inline int read(){
    int x=0,f=1;char c;
    do c=getchar(),f=c=='-'?-1:f; while(!isdigit(c));
    do x=(x<<3)+(x<<1)+c-'0',c=getchar(); while(isdigit(c));
    return x*f;
}
int n,m,Block_size,Block_num,x,y,k;
int a[N],b[N],block[N],v[1020];
char c[5];
inline void reset(int x){
    if(x==block[n]) return;
    for(int i=Block_size*(x-1)+1;i<=Block_size*x;i++)
        b[i]=a[i];
    sort(b+Block_size*(x-1)+1,b+Block_size*x+1);
}
inline void Add(int l,int r,int k){
    if(block[r]-block[l]<2){
        for(int i=l;i<=r;i++)
            a[i]+=k;
        reset(block[l]);reset(block[r]);
        return;
    }
    else{
        for(int i=block[l]+1;i<block[r];i++)
            v[i]+=k;
        for(int i=l;i<=block[l]*Block_size;i++)
            a[i]+=k;
        reset(block[l]);
        for(int i=r;i>=(block[r]-1)*Block_size+1;i--)
            a[i]+=k;
        reset(block[r]);
    }
}
inline int calc(int l,int r,int k){
    int tmp=0,x=r;
    while(l<=r){
        int mid=l+r>>1;
        if(b[mid]>=k) tmp=mid,r=mid-1;
        else l=mid+1;
    }
    if(!tmp) return 0;
    return x-tmp+1;
}
inline void Query(int l,int r,int k){
    int cnt=0;
    if(block[r]-block[l]<2){
        for(int i=l;i<=r;i++)
            if(a[i]+v[block[i]]>=k) cnt++;
        printf("%d\n",cnt);
        return;
    }
    else{
        for(int i=block[l]+1;i<block[r];i++){
            cnt+=calc(Block_size*(i-1)+1,Block_size*i,k-v[i]);
        }
        for(int i=l;i<=block[l]*Block_size;i++)
            if(a[i]+v[block[i]]>=k) cnt++;
        for(int i=r;i>=(block[r]-1)*Block_size+1;i--)
            if(a[i]+v[block[i]]>=k) cnt++;
        printf("%d\n",cnt);
    }
}
main(){
    n=read();m=read();
    Block_size=sqrt(n);
    for(int i=1;i<=n;i++){
        a[i]=read();
        block[i]=(i-1)/Block_size+1;
    }
    Block_num=block[n];
    for(int i=1;i<Block_num;i++)
        reset(i);
    while(m--){
        scanf("%s",c+1);
        x=read();y=read();k=read();
        switch(c[1]){
            case 'M': Add(x,y,k); break;
            case 'A': Query(x,y,k); break;
        }
    }
return 0;
}