Codevs_P1082 线段树练习3(线段树)

题目描述 Description

给你N个数，有两种操作：

1：给区间[a,b]的所有数增加X

2：询问区间[a,b]的数的和。

输入描述 Input Description

第一行一个正整数n，接下来n行n个整数，

再接下来一个正整数Q，每行表示操作的个数，

如果第一个数是1，后接3个正整数，

表示在区间[a,b]内每个数增加X,如果是2，

表示操作2询问区间[a,b]的和是多少。

输出描述 Output Description

对于每个询问输出一行一个答案

样例输入 Sample Input

3

1

2

3

2

1 2 3 2

2 2 3

样例输出 Sample Output

9

数据范围及提示 Data Size & Hint

数据范围

1<=n<=200000

1<=q<=200000

指针操作较慢，用f[]数组减少操作

#include<cstdio>
#include<iostream>
using namespace std;
#define N 200005
long long n,m,k,x,y,z;
long long d[N*3],f[N*3];
inline long long in(){
long long x=0;char ch=getchar();
while(ch>'9'||ch<'0') ch=getchar();
while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
return x;
}
void build(long long o,long long l,long long r){
if(l<r){
long long m=(l+r)>>1;
build(o*2,l,m);
build(o*2+1,m+1,r);
d[o]=d[o*2]+d[o*2+1];
}
else  d[o]=in();
}
void change(long long o,long long l,long long r,long long data){
if(x<=l&&r<=y){
f[o]+=data;
d[o]+=data*(r-l+1);
return;
}
long long m=(l+r)>>1;
if(x<=m) change(o*2,l,m,data);
if(m<y) change(o*2+1,m+1,r,data);
d[o]=d[o*2]+d[o*2+1]+f[o]*(r-l+1);
return;
}
long long sum(long long o,long long l,long long r,long long v){
long long m=(l+r)>>1,res=0;
if(x<=l&&r<=y) return d[o]+v*(r-l+1);
if(x<=m) res+=sum(o*2,l,m,f[o]+v);
if(y>m) res+=sum(o*2+1,m+1,r,f[o]+v);
return res;
}
int main(){
n=in();
build(1,1,n);
m=in();
for(int i=1;i<=m;i++){
k=in();
if(k==1){
x=in(),y=in(),z=in();
change(1,1,n,z);
}
else{
x=in(),y=in();
printf("%lld\n",sum(1,1,n,0));
}
}
return 0;
}