洛谷P2801 分块

题意：

区间加，查区间>= k k k的数的个数

思路：

区间大于等于k容易想到主席树，但是区间加，也无法标记永久化，所以难以维护。考虑数列分块

• 区间查询

  这里块贡献仍然独立，考虑块内维护有序序列，整块可以一次二分得到答案，散块暴力即可

  O ( n / B ∗ l o g B + B ) O(n/B*logB+B) O(n/B∗logB+B)

• 区间加

  对于整块，直接打 a d d add add标记即可，散块的话，暴力做，可以选 O ( B l o g B ) O(BlogB) O(BlogB)重构有序序列，但实际上我们发现可以通过记录 i d id id归并实现 O ( B + n / B ) O(B+n/B) O(B+n/B)

B = n l o g n B=\sqrt{nlogn} B=nlogn ​最优复杂度为 q n l o g n q\sqrt{nlogn} qnlogn ​，当然块大小还得适当调节一下

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+5;
int a[maxn],b[maxn],L[maxn],R[maxn],pos[maxn],add[maxn+10],num,n,q,p[maxn],c[maxn],cnt1,cnt2,big;
char s[2];
void change(int x,int l,int r){
    for(int i=L[x];i<=R[x];++i){
        if(p[i]>=l&&p[i]<=r)c[++cnt1]=a[p[i]];
        else c[++cnt2]=a[p[i]];
    }
    merge(c+1,c+cnt1+1,c+cnt1+1,c+cnt2+1,b+L[x]);
}
void update(int l,int r,int val){
    int pl=pos[l],pr=pos[r];
    cnt1=0,cnt2=r-l+1;
    if(pl==pr){
        for(int i=l;i<=r;++i)a[i]+=val;
        change(pl,l,r);
    }else{
        for(int i=pl+1;i<=pr-1;++i)add[i]+=val;
        for(int i=l;i<=R[pl];++i)a[i]+=val;
        cnt2=R[pl]-l+1;
        change(pl,l,r);
        cnt1=0;cnt2=r-L[pr]+1;
        for(int i=L[pr];i<=r;++i)a[i]+=val;
        change(pr,l,r);
    }
}
int query(int l,int r,int k){
    int ans=0;
    int pl=pos[l],pr=pos[r];
    if(pl==pr){
        for(int i=l;i<=r;++i)ans+=((a[i]+add[pl])>=k);
    }else{
        for(int i=l;i<=R[pl];++i)ans+=((a[i]+add[pl])>=k);
        for(int i=L[pr];i<=r;++i)ans+=((a[i]+add[pr])>=k);
        for(int i=pl+1;i<=pr-1;++i){
            ans+=(R[i]-(lower_bound(b+L[i],b+R[i]+1,k-add[i])-b)+1);
        }
    }
    return ans;
}
bool cmp(int x,int y){return a[x]<a[y];}
void init(){
    big=sqrt(n*log2(n)*1.65)+1;//防止分不了块 
    num=(n-1)/big+1;
    for(int i=1;i<=num;++i){
        L[i]=R[i-1]+1;
        R[i]=i*big;
    }
    R[num]=n;
    for(int i=1;i<=num;++i){
        for(int j=L[i];j<=R[i];++j){
            pos[j]=i;
        }
        sort(p+L[i],p+R[i]+1,cmp);
    }
    for(int i=1;i<=n;++i)b[i]=a[p[i]];
}
int main(){
    scanf("%d%d",&n,&q);
    for(int i=1;i<=n;++i)scanf("%d",&a[i]),p[i]=i;
    init();
    int l,r,x;
    for(int i=1;i<=q;++i){
        scanf("%s",s);
        scanf("%d%d%d",&l,&r,&x);
        if(s[0]=='A'){
            cout<<query(l,r,x)<<"\n";
        }else{
            update(l,r,x);
        }
    }
    return 0;
}

标签：pr,洛谷,分块,int,P2801,pos,add,maxn,pl
来源： https://blog.csdn.net/weixin_45539557/article/details/116462032