AcWing
  • 首页
  • 课程
  • 题库
  • 更多
    • 竞赛
    • 题解
    • 分享
    • 问答
    • 应用
    • 校园
  • 关闭
    历史记录
    清除记录
    猜你想搜
    AcWing热点
  • App
  • 登录/注册

树状数组

作者: 作者的头像   acwing_98726 ,  2025-05-09 17:15:06 · 河北 ,  所有人可见 ,  阅读 2


0


单点增加,范围查询

inline int lowbit(int x){
    return x&-x;
}

void add(int i,int v){
    while(i<=n){
        a[i]+=v;
        i+=lowbit(i);
    }
}

int query(int i){
    int res=0;
    while(i>0){
        res+=a[i];
        i-=lowbit(i);
    }
    return res;
}

单点查询,范围增加

存储原数组的差分数组即可

单点查询,范围增加

维护两个差分信息 d和(i-1)*d
sum=k*d(l,r)-(i-1)*d

二维范围查询,范围修改



int n,m;
int d1[2050][2050],d2[2050][2050],d3[2050][2050],d4[2050][2050];

inline int lowbit(int x){
    return x&-x;
}
void add(int x,int y,int v){
    int v1=v,v2=x*v,v3=y*v,v4=x*y*v;
    for(int i=x;i<=n;i+=lowbit(i)){
        for(int j=y;j<=m;j+=lowbit(j)){
            d1[i][j]+=v1;
            d2[i][j]+=v2;
            d3[i][j]+=v3;
            d4[i][j]+=v4;
        }
    }
}

void add(int a,int b,int c,int d,int v){
    add(a,b,v);
    add(a,d+1,-v);
    add(c+1,b,-v);
    add(c+1,d+1,v);
}

int query(int x,int y){
    int res=0;
    for(int i=x;i>0;i-=lowbit(i)){
        for(int j=y;j>0;j-=lowbit(j)){
            res+= (x + 1) * (y + 1) * d1[i][j] 
            - (y + 1) * d2[i][j] 
            - (x + 1) * d3[i][j] 
            + d4[i][j];
        }
    }
    return res;
}

int sum(int a, int b, int c, int d) {
    return query(c, d) - query(a - 1, d) - query(c, b - 1) + query(a - 1, b - 1);
}

0 评论

App 内打开
你确定删除吗?
1024
x

© 2018-2025 AcWing 版权所有  |  京ICP备2021015969号-2
用户协议  |  隐私政策  |  常见问题  |  联系我们
AcWing
请输入登录信息
更多登录方式: 微信图标 qq图标 qq图标
请输入绑定的邮箱地址
请输入注册信息