二维树状数组

1.单点更新,区间查询

int tree[N][N];  //行列分开看,每一行每一列都是一个一维树状数组
int n,m;         //n行m列
 
int lowbit(int x){return x&(-x);}
//单点更新,区间查询
void add(int x,int y,int val){  //单点更新
    while(x<=n){
        for(int i=y;i<=m;i+=lowbit(i)){  //列的一维树状数组
            tree[x][i]+=val;
        }
        x+=lowbit(x);
    }
}
int sum(int x,int y){  //返回(0,0,),(x,y)为对角顶点的矩阵和
    int res=0;
    while(x>0){
        for(int i=y;i>0;i-=lowbit(i)){
            res+=tree[x][i];
        }
        x-=lowbit(x);
    }
    return res;
}
int query(int x1,int y1,int x2,int y2){   //区间查询
    return sum(x2,y2)+sum(x2,y2)-sum(x2,y1-1)-sum(x1-1,y2);  //容斥,注意是否可能超longlong
}

2.区间更新,单点查询

//区间修改,单点查询,和一维树状数组差分思想一样,差分思想。
//二维前缀和:
//sum[i][j]=sum[i−1][j]+sum[i][j−1]−sum[i−1][j−1]+a[i][j]
//那么我们可以令差分数组d[i][j] 表示 a[i][j] 与 a[i−1][j]+a[i][j−1]−a[i−1][j−1] 的差。
 
void regionUpdate(int x1,int y1,int x2,int y2,int val){
    add(x1,y1,val);
    add(x2+1,y1,-val);
    add(x1,y2+1,-val);
    add(x2+1,y2+1,val);
}
int pointQuery(int x,int y){
return sum(x,y);

3.区间更新,区间查询

ll n, m, Q;
ll t1[N][N], t2[N][N], t3[N][N], t4[N][N];
void add(ll x, ll y, ll z){
    for(int X = x; X <= n; X += X & -X)
        for(int Y = y; Y <= m; Y += Y & -Y){
            t1[X][Y] += z;
            t2[X][Y] += z * x;
            t3[X][Y] += z * y;
            t4[X][Y] += z * x * y;
        }
}
void range_add(ll xa, ll ya, ll xb, ll yb, ll z){ //(xa, ya) 到 (xb, yb) 的矩形
    add(xa, ya, z);
    add(xa, yb + 1, -z);
    add(xb + 1, ya, -z);
    add(xb + 1, yb + 1, z);
}
ll ask(ll x, ll y){
    ll res = 0;
    for(int i = x; i; i -= i & -i)
        for(int j = y; j; j -= j & -j)
            res += (x + 1) * (y + 1) * t1[i][j]
                - (y + 1) * t2[i][j]
                - (x + 1) * t3[i][j]
                + t4[i][j];
    return res;
}
ll range_ask(ll xa, ll ya, ll xb, ll yb){
    return ask(xb, yb) - ask(xb, ya - 1) - ask(xa - 1, yb) + ask(xa - 1, ya - 1);
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值