链接:http://codevs.cn/problem/1082/
题目描述 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
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<string>
#include<vector>
#include<queue>
#include<set>
#include<map>
#include<cmath>
#define mst(ss,b) memset((ss),(b),sizeof(ss))
#define maxn 0x3f3f3f3f
using namespace std;
int n,m;
struct node{
int l,r;
long long v;
int flag;
} s[800001];
int a[200001];
void build(int root,int l,int r){
s[root].l=l;
s[root].r=r;
if(l==r){
s[root].v=a[l];
return ;
}
int mid=(l+r)/2;
build(root*2,l,mid);
build(root*2+1,mid+1,r);
s[root].v=s[root*2].v+s[root*2+1].v;
}
void putdown(int root){
int x=s[root].r-s[root].l+1;
s[root*2].flag+=s[root].flag;
s[root*2+1].flag+=s[root].flag;
s[root*2].v+=(x-(x>>1))*s[root].flag;
s[root*2+1].v+=s[root].flag*(x>>1);
s[root].flag=0;
}
void Up(int root,int l,int r,int v){
if(s[root].l==l && s[root].r==r){
s[root].flag+=v;
s[root].v+=(r-l+1)*v;
return ;
}
if(s[root].flag)
putdown(root);
int mid=(s[root].l+s[root].r)/2;
if(r<=mid){
Up(root*2,l,r,v);
}
else if(l>mid){
Up(root*2+1,l,r,v);
}
else{
Up(root*2,l,mid,v);
Up(root*2+1,mid+1,r,v);
}
s[root].v=s[root*2].v+s[root*2+1].v;
}
long long query(int root,int l,int r){
if(s[root].l==l && s[root].r==r){
return s[root].v;
}
if(s[root].flag)
putdown(root);
int mid=(s[root].l+s[root].r)/2;
if(r<=mid)
return query(root*2,l,r);
else if(l>mid)
return query(root*2+1,l,r);
else{
return query(root*2,l,mid)+query(root*2+1,mid+1,r);
}
}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
build(1,1,n);
int fl,x,y,z;
scanf("%d",&m);
for(int i=1;i<=m;i++){
scanf("%d",&fl);
if(fl==1){
scanf("%d%d%d",&x,&y,&z);
Up(1,x,y,z);
}
else{
scanf("%d%d",&x,&y);
printf("%lld\n",query(1,x,y));
}
}
return 0;
}