1. 程式人生 > 實用技巧 >線段樹模板1

線段樹模板1

如題,已知一個數列,你需要進行下面兩種操作:

  1. 將某區間每一個數加上kk。
  2. 求出某區間每一個數的和。

輸入格式

第一行包含兩個整數n, mn,m,分別表示該數列數字的個數和操作的總個數。

第二行包含nn個用空格分隔的整數,其中第ii個數字表示數列第ii項的初始值。

接下來mm行每行包含33或44個整數,表示一個操作,具體如下:

  1. 1x y k:將區間[x, y][x,y]內每個數加上kk。
  2. 2x y:輸出區間[x, y][x,y]內每個數的和。

輸出格式

輸出包含若干行整數,即為所有操作 2 的結果。

指標好啊。。

程式碼:

#include <cstdio>

using namespace std;

typedef long long int ll;
const int maxn=100005;
int n,q;
ll a[maxn*4];
struct Segment{
int l,r;
ll tag,v;
Segment *ls,*rs;
Segment(const int L,const int R){
l=L;r=R;
if(l==r){
tag=0;
v=a[l];
ls=rs=NULL;
}else{
tag=0;
int M=(L+R)>>1;
ls=new Segment(L,M);
rs=new Segment(M+1,R);
pushup();
}
}
inline void pushup(){
v=ls->v+rs->v;

}
inline void pushdown(){
if(tag==0)return;
ls->maketag(tag);
rs->maketag(tag);
tag=0;
}
inline void maketag(ll w){
v+=(r-l+1)*w;
tag+=w;
}
inline void upd(const int L,const int R,ll w){
if(inrange(L,R))maketag(w);
else if(!outrange(L,R)){
pushdown();
ls->upd(L,R,w);
rs->upd(L,R,w);
pushup();
}
}
inline ll qrt(const int L,const int R){
if(inrange(L,R))return v;
if(outrange(L,R))return 0;
pushdown();
return ls->qrt(L,R)+rs->qrt(L,R);
}
inline bool inrange(const int L,const int R){
return (l>=L)&&(r<=R);
}
inline bool outrange(const int L,const int R){
return (l>R)||(r<L);
}
};
int main(){
scanf("%d%d",&n,&q);
for(int i=1;i<=n;i++)scanf("%lld",a+i);
Segment *rot=new Segment(1,n);
for(ll o,x,y,z;q;q--){
scanf("%lld%lld%lld",&o,&x,&y);
if(o==1){
scanf("%lld",&z);
rot->upd(x,y,z);
}else{
printf("%lld\n",rot->qrt(x,y));
}
}
return 0;
}