【模板】CDQ分治

__stdcall大佬的讲解

这里贴的代码写的是点修改、区间查询的题。

 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #define LL long long
 #define rg register
 #define N 2000010
 #define mid ((l+r)>>1)
 using namespace std;
 int n,m,cnt,tot,ans[N];
 struct rec{
     int pos,val,type;
     //对于修改操作，val表示增量，对于查询操作，val表示它是第几个询问；
     //type为0表示修改，1表示区间左端点(开），2表示区间右端点
     bool operator<(const rec&x)const{return pos==x.pos?type<x.type:pos<x.pos;}
 }a[N],tmp[N];
 inline int read(){
     int k=,f=; char c=getchar();
     while(c<''||c>'')c=='-'&&(f=-),c=getchar();
     while(''<=c&&c<='')k=k*+c-'',c=getchar();
     return k*f;
 }
 void cdq(int l,int r){
     if(l>=r) return;
     cdq(l,mid),cdq(mid+,r);
     LL sum=; int p1=l,p2=mid+,cnt=;
     while(p1<=mid&&p2<=r){
         if(a[p1]<a[p2]){
             if(!a[p1].type) sum+=a[p1].val; tmp[++cnt]=a[p1++];
         }
         else{
             if(a[p2].type==) ans[a[p2].val]-=sum;
             else if(a[p2].type==) ans[a[p2].val]+=sum;
             tmp[++cnt]=a[p2++];
         }
     }
     for(rg int i=p1;i<=mid;i++) tmp[++cnt]=a[i];
     for(rg int i=p2;i<=r;i++){
         if(a[i].type==) ans[a[i].val]-=sum;
         else if(a[i].type==) ans[a[i].val]+=sum;
         tmp[++cnt]=a[i];
     }
     for(rg int i=l;i<=r;i++) a[i]=tmp[i-l+];
 }
 int main(){
     n=read(); m=read();
     for(rg int i=;i<=n;i++) a[++tot]=(rec){i,read(),};
     for(rg int i=;i<=m;i++){
         int opt=read(),x=read(),y=read();
         if(opt==) a[++tot]=(rec){x,y,};
         else a[++tot]=(rec){x-,++cnt,},a[++tot]=(rec){y,cnt,};
     }
     cdq(,tot);
     for(rg int i=;i<=cnt;i++) printf("%d\n",ans[i]);
 }