题目
题目描述
如题,已知一个数列,你需要进行下面两种操作:
- 将某区间每一个数加上 𝑘。
- 求出某区间每一个数的和。
输入格式
第一行包含两个整数 𝑛,𝑚,分别表示该数列数字的个数和操作的总个数。
第二行包含 𝑛 个用空格分隔的整数,其中第 𝑖 个数字表示数列第 𝑖 项的初始值。
接下来 𝑚 行每行包含 3 或 4 个整数,表示一个操作,具体如下:
1 x y k
:将区间 [𝑥,𝑦] 内每个数加上 𝑘。2 x y
:输出区间 [𝑥,𝑦] 内每个数的和。
输出格式
输出包含若干行整数,即为所有操作 2 的结果。
代码
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N=1e5+10;
int n,m,a[N],op;
struct node{
int l,r,sum,lazy;
}tree[4*N];
void pushup(int p){
tree[p].sum=tree[2*p].sum+tree[2*p+1].sum;
}
void build(int p,int l,int r){
tree[p].l=l;
tree[p].r=r;
tree[p].sum=tree[p].lazy=0;
if(l==r){
tree[p].sum=a[l];
return ;
}
int mid=(l+r)/2;
build(2*p,l,mid);
build(2*p+1,mid+1,r);
pushup(p);
}
void pushdown(int p){
tree[2*p].lazy+=tree[p].lazy;
tree[2*p+1].lazy+=tree[p].lazy;
tree[2*p].sum+=(tree[2*p].r-tree[2*p].l+1)*tree[p].lazy;
tree[2*p+1].sum+=(tree[2*p+1].r-tree[2*p+1].l+1)*tree[p].lazy;
tree[p].lazy=0;
}
void update(int p,int l,int r,int k){
if(tree[p].l>r||tree[p].r<l){
return ;
}
if(tree[p].l>=l&&tree[p].r<=r){
tree[p].sum+=(tree[p].r-tree[p].l+1)*k;
tree[p].lazy+=k;
return ;
}
if(tree[p].lazy) pushdown(p);
update(2*p,l,r,k);
update(2*p+1,l,r,k);
pushup(p);
}
int query(int p,int l,int r){
if(tree[p].l>r||tree[p].r<l){
return 0;
}
if(tree[p].l>=l&&tree[p].r<=r){
return tree[p].sum;
}
if(tree[p].lazy) pushdown(p);
return query(2*p,l,r)+query(2*p+1,l,r);
}
signed main(){
cin>>n>>m;
for(int i=1;i<=n;i++){
cin>>a[i];
}
build(1,1,n);
for(int i=1;i<=m;i++){
cin>>op;
int x,y,k;
cin>>x>>y;
if(op==1){
cin>>k;
update(1,x,y,k);
}
else{
cout<<query(1,x,y)<<endl;
}
}
return 0;
}