树状数组的区间更新和区间查询

A[1-n]的和

所以需要维护两个树状数组

sum1[i] = D[i]

sum2[i] = D[i]*(i-1)

然后修改相应的更新和求和方法即可

void update(){
    int x = i;
    while(i <= n){
        sum1[i] += k;
        sum2[i] += k * (x-1);
        i += -i&i;
    }
}
int sum(int i){
    int sum = 0,x = i;
    while( i<= n ){
        sum += x * sum1[i]-sum2[i];
        i -= -i&i;
    }
    return sum;
}

 

全部评论

相关推荐

不愿透露姓名的神秘牛友
06-05 15:27
点赞 评论 收藏
分享
流浪的神仙:无恶意,算法一般好像都得9硕才能干算法太卷啦
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务