区间修改维护差分序列。
区间查询就需要再推一个公式化简。
假设维护的差分序列为b[].
查询一段区间[l,r],可以用连段的差值相减,即求一段区间[l,r] = [1,r] - [1,l-1].
有[1,p]的值为
∑
i
=
1
p
∑
j
=
1
i
b
[
j
]
\sum_{i=1}^{p}\sum_{j=1}^{i}b[j]
∑i=1p∑j=1ib[j] , p 为l-1或者r
再次化简,得到
∑
i
=
1
p
(
p
−
i
+
1
)
b
[
i
]
\sum_{i=1}^{p}(p-i+1)b[i]
∑i=1p(p−i+1)b[i]
上式拆开得到
(
p
+
1
)
∑
i
=
1
p
b
[
i
]
−
∑
i
=
1
p
i
b
[
i
]
(p+1)\sum_{i=1}^{p}b[i]-\sum_{i=1}^{p}ib[i]
(p+1)∑i=1pb[i]−∑i=1pib[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;
}