二维线段树、二维树状数组模板

二维线段树模板

原理是树套树。

const int MAXN=maxv;
int N,M;//二维线段树的双维
//二维线段树模板，维护单点修改、区间查询和的操作
struct Seg1//内层线段树，维护y轴
{
    int sum[MAXN<<2];
    void build(int now,int l,int r){
        sum[now]=0;
        if(l==r) return;
        int mid=(l+r)>>1;
        build(now<<1,l,mid);
        build(now<<1|1,mid+1,r);
    }
    int Query(int now,int l,int r,int gl,int gr){
        if(l>=gl&&r<=gr) return sum[now];
        int mid=(l+r)>>1,sum=0;
        if(gl<=mid) sum+=Query(now<<1,l,mid,gl,gr);
        if(gr>mid) sum+=Query(now<<1|1,mid+1,r,gl,gr);
        return sum;
    }
    void Replace(int now,int l,int r,int y,int k){
        if(l==r&&r==y) {sum[now]+=k;return;}
        int mid=(l+r)>>1;
        if(y<=mid) Replace(now<<1,l,mid,y,k);
        if(y>mid) Replace(now<<1|1,mid+1,r,y,k);
        sum[now]=sum[now<<1]+sum[now<<1|1];
    }
};
struct Seg2//外层线段树，维护x轴
{
    Seg1 t[MAXN<<2];
    //gl、gr、dn、up分别为左右下上
    void build(int now,int l,int r){
        t[now].build(1,1,M);
        if(l==r) return;
        int mid=(l+r)>>1;
        build(now<<1,l,mid);
        build(now<<1|1,mid+1,r);
    }
    int Query(int now,int l,int r,int gl,int gr,int dn,int up){
        if(l>=gl&&r<=gr) return t[now].Query(1,1,M,dn,up);
        int mid=(l+r)>>1,sum=0;
        if(gl<=mid) sum+=Query(now<<1,l,mid,gl,gr,dn,up);
        if(gr>mid) sum+=Query(now<<1|1,mid+1,r,gl,gr,dn,up);
        return sum;
    }
    void Replace(int now,int l,int r,int x,int y,int k){
        t[now].Replace(1,1,M,y,k);
        if(l==r) return;
        int mid=(l+r)>>1;
        if(x<=mid) Replace(now<<1,l,mid,x,y,k);
        if(x>mid) Replace(now<<1|1,mid+1,r,x,y,k);
    }
}tree[MAXN];
N=1e3,M=1e3;
for(int i=1;i<=cnt;i++) tree[i].build(1,1,N);   //建树

二维树状数组模板

int N,M;//二维树状数组的双维
//二维树状数组模板，维护单点修改、区间查询和的操作
struct SecTree{
    int a[maxv][maxv];
    inline void add(int x,int y,int k){
        for(int i=x;i<=N;i+=i&-i) for(int j=y;j<=M;j+=j&-j) a[i][j]+=k;
    }
    inline int query(int x, int y) {
        int ans=0;
        for(int i=x;i;i-=i&-i) for(int j=y;j;j-=j&-j) ans+=a[i][j];
        return ans;
    }
    inline int Query(int x1, int y1,int x2,int y2) {
        return query(x2,y2)-query(x1-1,y2)-query(x2,y1-1)+query(x1-1,y1-1);
    }  
};
N=1e3,M=1e3;