题目连接 http://codevs.cn/problem/1082/
Description
给你N个数,有两种操作:
1:给区间[a,b]的所有数增加X
2:询问区间[a,b]的数的和。
Input
第一行一个正整数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 ouput
9
数据范围及提示
Data Size & Hint
数据范围
1<=n<=200000
1<=q<=200000
代码
#include <cstdio> #include <cmath> #include <cstring> #include <ctime> #include <iostream> #include <algorithm> #include <set> #include <vector> #include <queue> #include <typeinfo> #include <map> #include <stack> typedef long long ll; #define inf 0x7fffffff using namespace std; inline ll read() { ll x=0,f=1; char ch=getchar(); while(ch<'0'||ch>'9') { if(ch=='-')f=-1; ch=getchar(); } while(ch>='0'&&ch<='9') { x=x*10+ch-'0'; ch=getchar(); } return x*f; } //************************************************************************************** struct ss { int l,r; ll sum; int tag; }tr[4*200005]; int n; int a[200005]; void build(int k,int l,int r) { tr[k].l=l; tr[k].r=r; if(l==r) {tr[k].sum=a[l];return;} int mid=(l+r)>>1; build(k<<1,l,mid); build(k<<1|1,mid+1,r); tr[k].sum=tr[k<<1].sum+tr[k<<1|1].sum; } void pushdown(int k) { int x=tr[k].r-tr[k].l+1; tr[k<<1].tag+=tr[k].tag; tr[k<<1|1].tag+=tr[k].tag; tr[k<<1].sum+=(x-(x>>1))*tr[k].tag; tr[k<<1|1].sum+=(x>>1)*tr[k].tag; tr[k].tag=0; } void update(int k,int a,int b,int x) { if(a==tr[k].l&&b==tr[k].r) { tr[k].tag+=x; tr[k].sum+=(b-a+1)*x; return ; } if(tr[k].tag)pushdown(k); int mid=(tr[k].l+tr[k].r)>>1; if(b<=mid)update(k<<1,a,b,x); else if(a>mid)update(k<<1|1,a,b,x); else { update(k<<1,a,mid,x); update(k<<1|1,mid+1,b,x); } tr[k].sum=tr[k<<1].sum+tr[k<<1|1].sum; } ll ask(int k,int a,int b) { if(a==tr[k].l&&b==tr[k].r){return tr[k].sum;} if(tr[k].tag) pushdown(k); int mid=(tr[k].l+tr[k].r)>>1; if(b<=mid)return ask(k<<1,a,b); else if(a>mid) return ask(k<<1|1,a,b); else { return (ask(k<<1,a,mid)+ask(k<<1|1,mid+1,b)); } } int main() { scanf("%d",&n); for(int i=1;i<=n;i++) { scanf("%d",&a[i]); } build(1,1,n); int q; scanf("%d",&q); for(int i=1;i<=q;i++) { int x,aa,b,mm; scanf("%d",&mm); if(mm==1) { scanf("%d%d%d",&aa,&b,&x); update(1,aa,b,x); } else { scanf("%d%d",&aa,&b); printf("%lld\n",ask(1,aa,b)); } } return 0; }