AcWing 243. 一个简单的整数问题2-（区间修改,区间查询）-526互联

给定一个长度为 $N$ 的数列 $A$ ，以及 $M$ 条指令，每条指令可能是以下两种之一：

C l r d，表示把 $A [l], A [l + 1], \dots, A [r]$ 都加上 $d$ 。
Q l r，表示询问数列中第 $l \sim r$ 个数的和。

对于每个询问，输出一个整数表示答案。

输入格式

第一行两个整数 $N, M$ 。

第二行 $N$ 个整数 $A [i]$ 。

接下来 $M$ 行表示 $M$ 条指令，每条指令的格式如题目描述所示。

输出格式

对于每个询问，输出一个整数表示答案。

每个答案占一行。

数据范围

$1 \leq N, M \leq 10^{5}$
$| d | \leq 10000$ ,
$| A [i] | \leq 10^{9}$

输入样例：

10 5
1 2 3 4 5 6 7 8 9 10
Q 4 4
Q 1 10
Q 2 4
C 3 6 3
Q 2 4

输出样例：

代码实现：

#include<iostream>
#include<algorithm>
using namespace std;
#define int long long
const int N=1e5+5;
int a[N],tr1[N],tr2[N];
int n,m;
int lowbit(int x){
    return x&-x;
}
void add(int tr[],int x,int y){
    for(int i=x;i<=n;i+=lowbit(i))tr[i]+=y;
}
int query(int tr[],int x){
    int res=0;
    for(int i=x;i;i-=lowbit(i))res+=tr[i];
    return res;
}
int pre(int x){
    return (x+1)*query(tr1,x)-query(tr2,x);
}
signed main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        int x=a[i]-a[i-1];
        add(tr1,i,x);
        add(tr2,i,i*x);
    }
    while(m--){
        string s;
        cin>>s;
        int x,y,z;
        if(s[0]=='C'){
            cin>>x>>y>>z;
            add(tr1,x,z),add(tr2,x,x*z);
            add(tr1,y+1,-z),add(tr2,y+1,(y+1)*-z);
        }else{
            cin>>x>>y;
            cout<<pre(y)-pre(x-1)<<endl;
        }
    }
    return 0;
}