poj3468 线段树 延迟操作，修改树状数组

题意：区间更新，区间求值。

思路：

对于区间更新，我们可以用线段树的延迟操作。或者用两颗线段树来实现(logn)的区间更新。

关于延迟操作

1.带有延迟标记的区间更新。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=1e5+6;
int n,a[N],q;
typedef long long ll;
struct node{
   int l,r;
   ll sum,lazy;
   void update(ll x){
      sum+=1ll*(r-l+1)*x;
      lazy+=x;
   }
}tree[N*4];

void push_up(int x){
   tree[x].sum=tree[x<<1].sum+tree[x<<1|1].sum;
}

void push_down(int x){
  int lazyval=tree[x].lazy;
  if(lazyval){
    tree[x<<1].update(lazyval);
    tree[x<<1|1].update(lazyval);
    tree[x].lazy=0;
  }
}


void build(int x,int l,int r){
    tree[x].l=l,tree[x].r=r;
    tree[x].sum=0;tree[x].lazy=0;
    if(l==r){
        scanf("%lld",&tree[x].sum);
        return;
    }
    int mid=(l+r)/2;
    build(x<<1,l,mid);
    build(x<<1|1,mid+1,r);
    push_up(x);
}

void update(int x,int l,int r,int val){
   int L=tree[x].l,R=tree[x].r;
   if(l<=L&&r>=R){
      tree[x].update(val);
   }
   else{
      push_down(x);
      int mid=(L+R)/2;
      if(mid>=l) update(x<<1,l,r,val);
      if(r>mid) update(x<<1|1,l,r,val);
      push_up(x);
   }
}

ll query(int x,int l,int r){
   int L=tree[x].l,R=tree[x].r;
   if(l<=L&&r>=R){
      return tree[x].sum;
   }
   else{
      ll res=0;
      push_down(x);
      int mid=(L+R)/2;
      if(mid>=l) res+=query(x<<1,l,r);
      if(r>mid) res+=query(x<<1|1,l,r);
      push_up(x);
      return res;
   }
}


int main(){
   while(~scanf("%d %d",&n,&q)){
      build(1,1,n);
      for(int i=0;i<q;i++){
         char c[2];scanf("%s",c);
         if(c[0]=='Q'){
            int l,r;scanf("%d%d",&l,&r);
            printf("%lld\n",query(1,l,r));
         }
         else{
            int l,r,x;scanf("%d%d%d",&l,&r,&x);
            update(1,l,r,x);
         }
      }
   }
}

2.修改线段树

没有延迟标记的线段树只能对单个点高效更新（logn），对一个区域只对所有点更新。为了保持线段树高效，对于每个节点，我们考虑维护两个数据：
a.给这个节点对应的区间内所有元素共同加上的值
b.在这个节点对应的区间内除去a之外的和

那么对应求一个区间的和等于a*(r-l+1)+b。
同时维护这两个数据每一步都是（logn），实现了区间更新(logn).

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N=(1<<18)-1;
ll data[N],datb[N];

void add(int a,int b,int x,int k,int l,int r){
  if(a<=l&&r<=b) data[k]+=x;
  else if(l<b&&a<r){
     datb[k]+=(min(b,r)-max(a,l))*x;
     add(a,b,x,k*2+1,l,(l+r)/2);
     add(a,b,x,k*2+2,(l+r)/2,r);
  }
}

ll sum(int a,int b,int k,int l,int r){
    if(b<=l||r<=a) return 0;
    else if(a<=l&&r<=b){
        return data[k]*(r-l)+datb[k];
    }
    else{
        ll res=data[k]*(min(b,r)-max(a,l));
        res+=sum(a,b,k*2+1,l,(l+r)/2);
        res+=sum(a,b,k*2+2,(l+r)/2,r);
        return res;
    }
}

int main(){
     int q,n;
     while(~scanf("%d %d",&n,&q)){
        memset(data,0,sizeof(data));
        memset(datb,0,sizeof(datb));
        for(int i=0;i<n;i++){
            int x;scanf("%d",&x);
            add(i,i+1,x,0,0,n);
        }
        //cout<<sum(3,4,0,0,n)<<endl;
        for(int i=0;i<q;i++){
            //getchar();
            char c[2];scanf("%s",c);
            if(c[0]=='C'){
                int l,r,x;scanf("%d%d%d",&l,&r,&x);
                add(l-1,r,x,0,0,n);
            }
            else {
                int l,r;scanf("%d%d",&l,&r);
                printf("%lld\n",sum(l-1,r,0,0,n));
            }
        }
     }
}

3.修改树状数组

和线段树类似，树状数组也可以通过在每个节点上维护两个值，高效的经行区间更新。

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
typedef long long ll;
const int N=1000001;
ll b0[N+10],b1[N+10];

int lowbit(int x){ return x&(-x);}
void add(ll *b,int p,int x){
   while(p<N){
      b[p]+=x;
      p+=lowbit(p);
   }
}

ll sum(ll *b,int x){
   ll res=0;
   while(x>0){
     res+=b[x];
     x-=lowbit(x);
   }
   return res;
}

int main(){
    int n,q;
    while(~scanf("%d %d",&n,&q)){
        memset(b0,0,sizeof(b0));
        memset(b1,0,sizeof(b1));
        for(int i=1;i<=n;i++){
            int x;scanf("%d",&x);
            add(b0,i,x);
        }
        for(int i=0;i<q;i++){
           char c[2];scanf("%s",c);
           if(c[0]=='Q'){
              int l,r;scanf("%d %d",&l,&r);
              ll res=sum(b0,r)+sum(b1,r)*r;
              res-=(sum(b0,l-1)+sum(b1,l-1)*(l-1));
              printf("%lld\n",res);
           }
           else{
              int l,r,x;scanf("%d %d %d",&l,&r,&x);
              add(b0,l,-(l-1)*x);
              add(b0,r+1,x*r);
              add(b1,l,x);
              add(b1,r+1,-x);
           }
        }
    }
}