维护一个整数序列,支持2种操作:
1、修改:将给定区间中的每个数增加一个值;
2、查询:查询给定区间中所有数的和。
View Code
#include <stdio.h> #define N 100010 #define Node 800000 int n,m,x[N]; long long sum[N]; int left[Node],right[Node]; long long incsum[Node],inc[Node]; void create(int i,int l,int r) { left[i]=l; right[i]=r; inc[i]=0; incsum[i]=0; if(l==r) return; int mid=(l+r)>>1; create(2*i,l,mid); create(2*i+1,mid+1,r); } void update(int i,int l,int r,int x) { int mid=(left[i]+right[i])>>1; incsum[i]+=x*(r-l+1); if(left[i]==l && right[i]==r) { inc[i]+=x; return; } if(mid<l) { update(2*i+1,l,r,x); } else if(mid>=r) { update(2*i,l,r,x); } else { update(2*i,l,mid,x); update(2*i+1,mid+1,r,x); } } long long get(int i,int l,int r) { int mid=(left[i]+right[i])>>1; if(inc[i]) { inc[2*i]+=inc[i]; inc[2*i+1]+=inc[i]; incsum[2*i]+=inc[i]*(mid-left[i]+1); incsum[2*i+1]+=inc[i]*(right[i]-mid); inc[i]=0; } if(l==left[i] && r==right[i]) return incsum[i]; if(mid<l) return get(2*i+1,l,r); if(mid>=r) return get(2*i,l,r); return get(2*i,l,mid)+get(2*i+1,mid+1,r); } int main() { int i,a,b,c; char ch; while(~scanf("%d%d",&n,&m)) { create(1,1,n); sum[0]=0; for(i=1;i<=n;i++) scanf("%d",&x[i]),sum[i]=sum[i-1]+x[i]; for(i=1;i<=m;i++) { ch=0; while(ch!='Q' && ch!='C') scanf("%c",&ch); if(ch=='Q') { scanf("%d%d",&a,&b); printf("%lld\n",sum[b]-sum[a-1]+get(1,a,b)); } else { scanf("%d%d%d",&a,&b,&c); update(1,a,b,c); } } } return 0; }