一个简单的整数问题2（acwing）（数状数组）

题目地址
配套地址

区间修改维护差分序列。
区间查询就需要再推一个公式化简。

假设维护的差分序列为b[].
查询一段区间[l,r]，可以用连段的差值相减，即求一段区间[l,r] = [1,r] - [1,l-1].

有[1,p]的值为 ${\sum }_{i=1}^p{\sum }_{j=1}^{i}b[j]$ , p 为l-1或者r
再次化简，得到${\sum }_{i=1}^p(p-i+1)b[i]$

上式拆开得到 $(p+1){\sum }_{i=1}^{p}b[i]-{\sum }_{i=1}^{p}ib[i]$
所以我们维护两个前缀b[i] 和 i*b[i]。

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N = 1e5 + 10;

int a[N],b[N];
int c1[N],c2[N];
int n,m;

#define lowbit(x) ((x) & (-x))

void change(int x,int v){
    for(int i=x;i<=n;i+=lowbit(i)) c1[i] += v,c2[i] += x*v;
}

int ask(int x){
    int res = 0;
    for(int i=x;i;i-=lowbit(i)) res += (x + 1)*c1[i] - c2[i];
    return res;
}

signed main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>a[i],b[i] = a[i] - a[i-1];
    for(int i=1;i<=n;i++) change(i,b[i]);
    for(int i=1;i<=m;i++){
        string op; cin>>op;
        if(op == "Q"){
            int l,r; cin>>l>>r;
            cout<<ask(r)-ask(l-1)<<endl;
        }else{
            int l,r,d; cin>>l>>r>>d;
            change(l,d);change(r+1,-d);
        }
    }

    return 0;
}