【luogu 3372】【模板】線段樹1
阿新 • • 發佈:2017-10-15
-- 含義 return pri 一行 一個數 code update upload
輸出樣例#1:
題目描述
如題,已知一個數列,你需要進行下面兩種操作:
1.將某區間每一個數加上x
2.求出某區間每一個數的和
輸入輸出格式
輸入格式:
第一行包含兩個整數N、M,分別表示該數列數字的個數和操作的總個數。
第二行包含N個用空格分隔的整數,其中第i個數字表示數列第i項的初始值。
接下來M行每行包含3或4個整數,表示一個操作,具體如下:
操作1: 格式:1 x y k 含義:將區間[x,y]內每個數加上k
操作2: 格式:2 x y 含義:輸出區間[x,y]內每個數的和
輸出格式:
輸出包含若幹行整數,即為所有操作2的結果。
輸入輸出樣例
輸入樣例#1:5 5 1 5 4 2 3 2 2 4 1 2 3 2 2 3 4 1 1 5 1 2 1 4
11 8 20
說明
時空限制:1000ms,128M
數據規模:
對於30%的數據:N<=8,M<=10
對於70%的數據:N<=1000,M<=10000
對於100%的數據:N<=100000,M<=100000
(數據已經過加強^_^,保證在int64/long long數據範圍內)
樣例說明:
1 #include<cstdio> 2 #include<iostream> 3 #include<cstring> 4 #include<algorithm> 5 #definell l,mid,rt<<1 6 #define rr mid+1,r,rt<<1|1 7 #define LL long long 8 #define maxn 100005 9 using namespace std; 10 LL add[maxn<<2],sum[maxn<<2]; 11 void pushdown(int rt,int m){ 12 if(add[rt]){ 13 add[rt<<1]+=add[rt]; 14 add[rt<<1|1]+=add[rt];15 sum[rt<<1]+=add[rt]*(m-(m>>1)); 16 sum[rt<<1|1]+=add[rt]*(m>>1); 17 add[rt]=0; 18 } 19 } 20 void build(int l,int r,int rt){ 21 add[rt]=0; 22 if(l==r){ 23 scanf("%lld",&sum[rt]); 24 return ; 25 } 26 int mid=(l+r)>>1; 27 build(ll);build(rr); 28 sum[rt]=sum[rt<<1]+sum[rt<<1|1]; 29 } 30 void update(int ql,int qr,int c,int l,int r,int rt){ 31 if(ql<=l&&r<=qr){ 32 add[rt]+=c; 33 sum[rt]+=(LL)c*(r-l+1); 34 return ; 35 } 36 pushdown(rt,r-l+1); 37 int mid=(l+r)>>1; 38 if(ql<=mid) update(ql,qr,c,ll); 39 if(mid<qr) update(ql,qr,c,rr); 40 sum[rt]=sum[rt<<1]+sum[rt<<1|1]; 41 } 42 LL query(int ql,int qr,int l,int r,int rt){ 43 if(ql<=l&&r<=qr) return sum[rt]; 44 pushdown(rt,r-l+1); 45 int mid=(l+r)>>1; 46 LL ret=0; 47 if(ql<=mid) ret+=query(ql,qr,ll); 48 if(mid<qr) ret+=query(ql,qr,rr); 49 return ret; 50 } 51 int main(){ 52 int n,q; 53 scanf("%d%d",&n,&q); 54 build(1,n,1); 55 while(q--){ 56 int Q; 57 int a,b,c; 58 scanf("%d",&Q); 59 if(Q==2){ 60 scanf("%d%d",&a,&b); 61 printf("%lld\n",query(a,b,1,n,1)); 62 } 63 else{ 64 scanf("%d%d%d",&a,&b,&c); 65 update(a,b,c,1,n,1); 66 } 67 } 68 return 0; 69 }
【luogu 3372】【模板】線段樹1