POJ - 3468 A Simple Problem with Integers

题目大意：
给你n个数
进行m次操作
每次操作可以让一段区间加上一个数
也可以询问一段区间的和

思路：
树状数组
普通的树状数组只能进行单点查询，区间修改或者区间查询，单点修改的操作
那么我们如果要区间修改，区间查询该怎么做呢
我们可以考虑用差分来进行操作
假设c[i]=a[i]-a[i-1]
因为a[1]+a[2]+······=c[1]+c[1]+c[2]+······
所以Σai=n×c1+(n-1)×c2+······
得Σai=n×Σci-Σ(ci×(i-1))
那么我们就可以用两个数组
一个储存前n×ci的和,一个储存前ci×(i-1)的和
即可得出结果

AC代码:

#include <iostream>

using namespace std;
typedef long long ll;
const int N=100010;
ll sum1[N];
ll sum2[N];
ll a[N];
int lower_bit(int x){
      return x&(-x);
}
int n,m;
void add(int i,ll num){
      int x=i;//x是不变的
      while(i<=n){
        sum1[i]+=num;
        sum2[i]+=num*(x-1);//因为存储的是和后面的全部都要加上这个数
        i+=lower_bit(i);
      }
      return;
}
ll getsum(int i){
     ll sum=0;
     int x=i;
     while(i>0){
        sum+=x*sum1[i]-sum2[i];
        i-=lower_bit(i);
     }
     return sum;
}
int main()
{
    ios::sync_with_stdio(false);
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        add(i,a[i]-a[i-1]);
    }
    while(m--){
        char s;
        cin>>s;
        if(s=='Q'){
            int x,y;
        cin>>x>>y;
            cout<<getsum(y)-getsum(x-1)<<endl;
        }else{
            int x,y,k;
            cin>>x>>y>>k;
            add(x,k);//差分
            add(y+1,-k);
        };
    }
    return 0;
}