静态维护区间加等差数列的求和问题

原理:

维护一个数组,先进行m次操作,然后查询每个位置的值,每个操作给定四个参数l,r,a,k表示从l到r依次加上一个首项为a,公差为k的等差数列。
维护\(d_2\)数组,表示原数组的二阶差分。

修改操作

void add(int l,int r,int a,int k){
    d2[l]+=a;
    d2[l+1]+=k-a;
    d2[r+1]-=(r-l+1)*k+a;
    d2[r+2]-=(l-r)*k-a;
}

前缀和

void pre_sum(){
    for(int i=1;i<=n;++i){
        d2[i]+=d2[i-1];
    }
}

还原操作

pre_sum();
pre_sum();

模板

#include<bits/stdc++.h>
using namespace std;
const int MAXN=100005;
int n,m,d2[MAXN],l,r,a,k;
void add(int l,int r,int a,int k){
    d2[l]+=a;
    d2[l+1]+=k-a;
    d2[r+1]-=(r-l+1)*k+a;
    d2[r+2]-=(l-r)*k-a;
}
void pre_sum(){
    for(int i=1;i<=n;++i){
        d2[i]+=d2[i-1];
    }
}
int main(){
    scanf("%d %d",&n,&m);
    for(int i=1;i<=m;++i){
        scanf("%d %d %d %d",&l,&r,&a,&k);
        add(l,r,a,k);
    }
    pre_sum();
    pre_sum();
    for(int i=1;i<=n;++i){
        printf("%d%c",d2[i],i==n?'\n':' ');
    }
    return 0;
}

转载自:牛客:树状数组维护前缀和的前缀和

posted @ 2020-03-19 15:35  0x4f  阅读(205)  评论(0编辑  收藏  举报