LUGOU P3374 【模板】树状数组 1(CDQ 分治)

传送门

拿个二维偏序练练cdq板子，其实就和归并排序差不多，复杂度不太会，似乎nlogn?。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>

using namespace std;
const int MAXN = ;
const int MAXQ = MAXN*;
typedef long long LL;

inline LL rd(){
    LL x=,f=;char ch=getchar();
    while(!isdigit(ch)) {f=ch=='-'?:;ch=getchar();}
    while(isdigit(ch))  {x=(x<<)+(x<<)+ch-'';ch=getchar();}
    return f?x:-x;
}

int n,m,cnt,num;
LL ans[MAXN];

struct Query{
    int type,id;LL val;
    friend bool operator<(const Query A,const Query B){
        return A.id==B.id?A.type<B.type:A.id<B.id;
    }
}q[MAXQ],tmp[MAXQ];

void cdq(int l,int r){
    if(l==r) return;
    int mid=l+r>>;cdq(l,mid),cdq(mid+,r);
    int L=l,R=mid+;LL sum=;int o=;
    while(L<=mid && R<=r){
        if(q[L]<q[R]){
            if(q[L].type==) sum+=q[L].val;
            tmp[++o]=q[L++];
        }
        else{
            if(q[R].type==) ans[q[R].val]-=sum;
            else if(q[R].type==) ans[q[R].val]+=sum;
            tmp[++o]=q[R++];
        }
    }
    while(L<=mid) tmp[++o]=q[L++];
    while(R<=r) {
        if(q[R].type==) ans[q[R].val]-=sum;
        else if(q[R].type==) ans[q[R].val]+=sum;
        tmp[++o]=q[R++];
    }
    for(int i=;i<=o;i++) q[i+l-]=tmp[i];
}

int main(){
    n=rd(),m=rd();
    for(int i=;i<=n;i++)
        q[++cnt].id=i,q[cnt].type=,q[cnt].val=rd();
    for(int i=;i<=m;i++){
        q[++cnt].type=rd();
        if(q[cnt].type==) q[cnt].id=rd(),q[cnt].val=rd();
        else{
            q[cnt].id=rd()-;q[cnt].val=++num;
            q[++cnt].type=;q[cnt].id=rd();q[cnt].val=num;
        }
    }
//    cout<<"  "<<endl;
//    for(int i=1;i<=cnt;i++) {
//        cout<<q[i].type<<" "<<q[i].id<<" "<<q[i].val<<endl;
//    }
    cdq(,cnt);
    for(int i=;i<=num;i++) printf("%lld\n",ans[i]);
    return ;
}