树状数组区间维护及单点查询


树状数组区间维护

d[i]表示a[i]-a[i-1]的值

原理:http://wenku.baidu.com/link?url=WAxUY51ZHipbib4RyxMqhoujjcehIkWSTBxFF1Z8sjCToih7npdGhCg_HW2oKLqH3vRsQEO_QAUVYi7l6WPGFYV4WAMaWxeSPfFg4tIDXfe

#include<cstdio>

#define M 400500

using namespace std;

long long n,a[M],d[M],c1[M],c2[M],m,k,x,y,l,r;

long long lowbit(long long i){

 return i&(-i);

}

void update(long long l,long long x){

 long long k=l;

 while(k<=n){

  c1[k]+=x;

  c2[k]+=x*l;

  k+=lowbit(k);

 }

 return ;

}

long long sum(long long i){

 long long s=0,s1=0,s2=0,p=i;

 while(p>0){

  s1+=c1[p];

  s2+=c2[p];

  p-=lowbit(p);

 }

 s=s1*(i+1)-s2;

 return s;

}

int main(){

 scanf("%I64d",&n);

 for(int i=1;i<=n;i++){

  scanf("%I64d",&a[i]);

  d[i]=a[i]-a[i-1];

  for(int j=i-lowbit(i)+1;j<=i;j++){

   c1[i]+=d[j];

   c2[i]+=d[j]*j;

  }

 }

 scanf("%I64d",&m);

 for(int i=1;i<=m;i++){

  scanf("%I64d",&k);

  if (k==1){

   scanf("%I64d%I64d%I64d",&l,&r,&x);

   d[l]+=x,d[r+1]-=x;

   update(l,x);

   update(r+1,-x);

  }

  else{

   scanf("%I64d%I64d",&l,&r);

   printf("%I64d\n",sum(r)-sum(l-1));//查询点i即输出sum(i)-sum(i-1)即可

  }

 }

 return 0;

}