前置知识:
- 线段树是一种支持O(logn)区间操作的数据结构
- 具体支持的操作:区间增加/减少/乘以 某个值;查询区间总和(复杂度均为O(logn))
联系:
- 主席树需要使用动态开点的思想,而在其他题目中也很常用动态开点线段树而不是普通线段树.
模板:https://www.luogu.org/problemnew/show/P3372
#include<cstdio>
#include<iostream>
using namespace std;
const int N=300010;
struct node{
int ls,rs,lazy;
long long sum;
}tr[N];
int root=0,cnt=0;
void insert(int &root,int l,int r,int ll,int rr,int x){
if(!root)root=++cnt;
int b=min(r,rr)-max(l,ll)+1;
tr[root].sum+=b*x;
if(l>=ll&&r<=rr){
tr[root].lazy+=x;
return;
}
int mid=l+r>>1;
if(ll<=mid)insert(tr[root].ls,l,mid,ll,rr,x);
if(rr>mid)insert(tr[root].rs,mid+1,r,ll,rr,x);
}
long long query(int root,int l,int r,int ll,int rr){
if(l>=ll&&r<=rr)return tr[root].sum;
int mid=l+r>>1;
if(tr[root].lazy){
if(!tr[root].ls)tr[root].ls=++cnt;
tr[tr[root].ls].lazy+=tr[root].lazy;
tr[tr[root].ls].sum+=tr[root].lazy*(mid-l+1);
if(!tr[root].rs)tr[root].rs=++cnt;
tr[tr[root].rs].lazy+=tr[root].lazy;
tr[tr[root].rs].sum+=tr[root].lazy*(r-mid);
tr[root].lazy=0;
}
long long ans=0;
if(ll<=mid)ans+=query(tr[root].ls,l,mid,ll,rr);
if(rr>mid)ans+=query(tr[root].rs,mid+1,r,ll,rr);
return ans;
}
int main(){
int n,m;
cin>>n>>m;
for(int i=1;i<=n;i++){
int temp;
cin>>temp;
insert(root,1,n,i,i,temp);
}
for(int i=1;i<=m;i++){
int ta,tb,tc,td;
cin>>ta>>tb>>tc;
if(ta==1){
cin>>td;
insert(root,1,n,tb,tc,td);
}else if(ta==2){
cout<<query(root,1,n,tb,tc)<<endl;
}
}
return 0;
}