POJ-3468 A Simple Problem with Integers

http://poj.org/problem?id=3468

第一次手写的线段树，以后就用这个做模板好了。

自己风格的线段树：（区段更新

#define maxn 100005
#define ls(p) p<<1        //p左子结点下标
#define rs(p) p<<1|1    //p右子结点下标
struct segTree{
    int l,r;            //线段左右端点值
    int v;                //区间和
    int f;                //lazy标记
}t[maxn<<2];
void pushUp(int p){
    t[p].v=t[ls(p)].v+t[rs(p)].v;
}
void pushDown(int p){
    if(t[p].f){
        t[ls(p)].f+=t[p].f;
        t[ls(p)].v+=t[p].f*(t[ls(p)].r-t[ls(p)].l+1);
        t[rs(p)].f+=t[p].f;
        t[rs(p)].v+=t[p].f*(t[rs(p)].r-t[rs(p)].l+1);
        t[p].f=0;
    }
}
void Build(int p,int l,int r){
    t[p].l=l;
    t[p].r=r;
    t[p].f=0;
    if(l==r){
        //init : scanf("%d",&t[p].v);
        return ;
    }
    int m=(l+r)/2;
    Build(ls(p),l,m);
    Build(rs(p),m+1,r);
    pushUp(p);
}
void upDate(int p,int L,int R,int v){
    int l=t[p].l,r=t[p].r;
    if(l>=L && R>=r){
        t[p].f+=v;
        t[p].v+=(r-l+1)*v;
        return ;
    }
    pushDown(p);
    int m=(l+r)/2;
    if(L<=m) upDate(ls(p),L,R,v);
    if(R>m) upDate(rs(p),L,R,v);
    pushUp(p);
}
int Query(int p,int L,int R){
    int l=t[p].l,r=t[p].r;
    if(l>=L && R>=r)
        return t[p].v;
    pushDown(p);
    int ans=0;
    int m=(l+r)/2;
    if(L<=m) ans+=Query(ls(p),L,R);
    if(R>m) ans+=Query(rs(p),L,R);
    return ans;
}

 单点更新：

#define maxn 100005
#define ls(p) p<<1
#define rs(p) p<<1|1
struct segTree{
    int l,r;
    int v;
}t[maxn<<2];
void pushUp(int p){
    t[p].v=t[ls(p)].v+t[rs(p)].v;
}
void Build(int l,int r,int p){
    t[p].l=l;
    t[p].r=r;
    if(l==r){
        scanf("%d",&t[p].v);
        return;
    }
    int m=(l+r)/2;
    Build(l,m,ls(p));
    Build(m+1,r,rs(p));
    pushUp(p);
}
void upDate(int p,int d,int v){
    int l=t[p].l,r=t[p].r;
    if(l==r){
        t[p].v+=v;
        return;
    }
    int m=(l+r)/2;
    if(d<=m) upDate(ls(p),d,v);
    else upDate(rs(p),d,v);
    pushUp(p);
}
int Query(int p,int L,int R){
    int l=t[p].l,r=t[p].r;
    if(L<=l && r<=R)
        return t[p].v;
    int m=(l+r)/2;
    int ans=0;
    if(L<=m) ans+=Query(ls(p),L,R);
    if(R>m) ans+=Query(rs(p),L,R);
    return ans;
}

ac代码：

#include<cstdio>
#include<iostream>
using namespace std;
#define maxn 100005
#define ls(p) p<<1
#define rs(p) p<<1|1
struct segTree{
    int l,r;
    long long v;
    long long f;
}t[maxn<<2];
void pushUp(int p){
    t[p].v=t[ls(p)].v+t[rs(p)].v;
}
void pushDown(int p){
    if(t[p].f){
        t[ls(p)].f+=t[p].f;
        t[ls(p)].v+=t[p].f*(t[ls(p)].r-t[ls(p)].l+1);
        t[rs(p)].f+=t[p].f;
        t[rs(p)].v+=t[p].f*(t[rs(p)].r-t[rs(p)].l+1);
        t[p].f=0;
    }
}
void Build(int p,int l,int r){
    t[p].l=l;
    t[p].r=r;
    t[p].f=0;
    if(l==r){
        //init
        scanf("%lld",&t[p].v);
        return ;
    }
    int m=(l+r)/2;
    Build(ls(p),l,m);
    Build(rs(p),m+1,r);
    pushUp(p);
}
void upDate(int p,int L,int R,int v){
    int l=t[p].l,r=t[p].r;
    if(l>=L && R>=r){
        t[p].f+=v;
        t[p].v+=(r-l+1)*v;
        return ;
    }
    pushDown(p);
    int m=(l+r)/2;
    if(L<=m) upDate(ls(p),L,R,v);
    if(R>m) upDate(rs(p),L,R,v);
    pushUp(p);
}
long long Query(int p,int L,int R){
    int l=t[p].l,r=t[p].r;
    if(l>=L && R>=r)
        return t[p].v;
    pushDown(p);
    long long ans=0;
    int m=(l+r)/2;
    if(L<=m) ans+=Query(ls(p),L,R);
    if(R>m) ans+=Query(rs(p),L,R);
    return ans;
}
int main(){
    int N,Q;
    while(~scanf("%d%d",&N,&Q)){
        Build(1,1,N);
        while(Q--){
            char op[2];
            scanf("%s",op);
            if(op[0]=='Q'){
                int a,b;
                scanf("%d%d",&a,&b);
                printf("%lld
",Query(1,a,b));
            }else{
                int a,b,v;
                scanf("%d%d%d",&a,&b,&v);
                upDate(1,a,b,v);
            }
        }
    }
    return 0;
}

最近发现树状数组也能做区间更新+区间询问，也是炫酷。。。有必要收藏一下：

详见：http://blog.csdn.net/q573290534/article/details/6664454

#include<map>
#include<set>
#include<list>
#include<cmath>
#include<ctime>
#include<queue>
#include<stack>
#include<cctype>
#include<cstdio>
#include<string>
#include<vector>
#include<cstdlib>
#include<cstring>
#include<complex>
#include<utility>    //pair
#include<iostream>
#include<algorithm>
#define MAXN 100005
#define INF 0x3f3f3f3f
#define LL long long
#define DBL double
#define EPS 1e-6
#define PI acos(-1.0)
#define Test() cout<<"Test"<<endl;
#define Debug(a) cout<<#a<<" = "<<a<<endl;
#define Debug2(a,b) cout<<#a<<" = "<<a<<" , "<<#b<<" = "<<b<<endl;
using namespace std;

LL a[MAXN], b[MAXN], c[MAXN], n, m;

void addB(int p, LL v){
    for(int i=p; i; i-=i&(-i))
        b[i]+=v;
}
LL sumB(int p){
    LL s=0;
    for(int i=p; i<=n; i+=i&(-i))
        s+=b[i];
    return s;
}
void addC(int p, LL v){
    for(int i=p; i<=n; i+=i&(-i))
        c[i]+=p*v;        //p*v
}
LL sumC(int p){
    LL s=0;
    for(int i=p; i; i-=i&(-i))
        s+=c[i];
    return s;
}
LL query(int x){
    return x? sumB(x)*x+sumC(x-1): 0;
}

// poj 3468
int main(){
    while(cin >>n >>m){
        a[0]=0;
        for(int i=1, t; i<=n; i++){    
            scanf("%d", &t);
            a[i] = a[i-1]+t;
        }
        memset(b, 0, sizeof(b));
        memset(c, 0, sizeof(c));
        while(m--){
            char cmd[5];
            int l, r, v;
            scanf("%s", cmd);
            if(cmd[0] == 'Q'){
                scanf("%d%d", &l, &r);
                printf("%lld
", query(r)-query(l-1)+a[r]-a[l-1]);
            }else{
                scanf("%d%d%d", &l, &r, &v);
                addB(r, v); addC(r, v);
                if(l>1)
                { addB(l-1, -v); addC(l-1, -v); }
            }
            
        }
    }
    return 0;
}