- 一个简单的整数问题2
给定一个长度为 N 的数列 A,以及 M 条指令,每条指令可能是以下两种之一:
C l r d,表示把 A[l],A[l+1],…,A[r] 都加上 d。
Q l r,表示询问数列中第 l∼r 个数的和。
对于每个询问,输出一个整数表示答案。
输入格式
第一行两个整数 N,M。
第二行 N 个整数 A[i]。
接下来 M 行表示 M 条指令,每条指令的格式如题目描述所示。
输出格式
对于每个询问,输出一个整数表示答案。
每个答案占一行。
数据范围
1≤N,M≤105,
|d|≤10000,
|A[i]|≤109
输入样例:
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
输出样例:
4
55
9
15
本题使用了差分与前缀和的性质,关系如上图,开两个数组分别维护 b[i] 与 i*b[i]来进行区间查询与修改。树状数组写法
#include<cstdio>
using namespace std;
#define lowbit(x)x&-x
typedef long long ll;
const int N = 1e5+10;
ll tr1[N],tr2[N],n,m,a[N];
inline void update(ll tr[],int u,ll x){for(;u<=n;u+=lowbit(u))tr[u]+=x;}
inline ll sum(ll tr[],int u){ll res(0);for(;u;u-=lowbit(u))res+=tr[u];return res;}
inline ll get(int u){return sum(tr1,u)*(u+1)-sum(tr2,u);}
int main(){
scanf("%d%d",&n,&m);
for(int i(1);i<=n;i++){
scanf("%d",&a[i]);
}
for(int i(1);i<=n;i++){
int b=a[i]-a[i-1];
update(tr1,i,b);
update(tr2,i,(ll)b*i);
}
while(m--){
int l,r,d;
char op[2];
scanf("%s %d %d",op,&l,&r);
if(*op=='Q'){
printf("%lld\n",get(r)-get(l-1));
}
else {
scanf("%d",&d);
update(tr1,l,d),update(tr1,r+1,-d);
update(tr2,l,(ll)d*l),update(tr2,r+1,(ll)(r+1)*-d);
}
}
}
线段树写法,加入add标记
#include<cstdio>
#pragma optimize(3)
using namespace std;
typedef long long ll;
inline ll lowbit(ll x){return x&-x;}
const int N = 1e5+10;
struct xds{
int l,r;
ll sum,add;
}tr[N*4];
int m,n,a[N];
inline void pushdown(int u){
auto &root=tr[u],&l=tr[u<<1],&r=tr[u<<1|1];
if(root.add){
l.add+=root.add,l.sum+=(ll)(l.r-l.l+1)*root.add;
r.add+=root.add,r.sum+=(ll)(r.r-r.l+1)*root.add;
root.add=0;
}
}
inline void pushup(xds&u,xds&l,xds&r){
u.sum=l.sum+r.sum;
}
inline void pushup(int u){
pushup(tr[u],tr[u<<1],tr[u<<1|1]);
}
inline void build(int u,int l,int r){
if(l==r){
tr[u]={l,r,a[r],0};
}
else {
tr[u]={l,r};
int mid=l+r>>1;
build(u<<1,l,mid),build(u<<1|1,mid+1,r);
pushup(u);
}
}
inline ll query(int u,int l,int r){
if(tr[u].l>=l&&tr[u].r<=r)return tr[u].sum;
else {
pushdown(u);
int mid=tr[u].l+tr[u].r>>1;
ll sum=0;
if(l<=mid)sum+=query(u<<1,l,r);
if(r>mid)sum+=query(u<<1|1,l,r);
return sum;
}
}
inline void modify(int u,int l,int r,int d){
if(tr[u].l>=l&&tr[u].r<=r){
tr[u].sum+=(ll)(tr[u].r-tr[u].l+1)*d;
tr[u].add+=d;
}
else {
int mid=tr[u].l+tr[u].r>>1;
pushdown(u);
if(l<=mid)modify(u<<1,l,r,d);
if(r>mid)modify(u<<1|1,l,r,d);
pushup(u);
}
}
int main(){
scanf("%d%d",&n,&m);
for(int i(1);i<=n;i++)scanf("%d",&a[i]);
build(1,1,n);
while(m--){
char op[2];
int l,r,x;
scanf("%s %d %d",op,&l,&r);
if(*op=='Q'){
printf("%lld\n",query(1,l,r));
}
else {
scanf("%d",&x);
modify(1,l,r,x);
}
}
}