线段树模板

首先感谢以下两篇博客：

http://blog.csdn.net/metalseed/article/details/8039326

http://blog.csdn.net/acmer_ak/article/details/52002537

第一篇博客讲述了很多线段树的题目以及解答（很不错的模板），第二篇博客讲述了一个很好的思路。

一、以poj 3468 为原型的一份模板：成段更新、区间求和、（区间最值/gcd），后期会更新模板...

#include <cstdio>
#include <iostream>
#include <climits>
#define maxn 100000+1000
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
using namespace std;


typedef long long ll;
int a[maxn];
struct node{
    int l,r;
    ll sum,mark;
    int mmax;
}sgt[maxn<<2];
/***
更新父节点
***/
void pushUp(int rt){
    sgt[rt].sum = (sgt[rt << 1].sum + sgt[rt<<1|1].sum);
    //sgt[rt].mmax = max(sgt[rt << 1].mmax , sgt[rt<<1|1].mmax);
}
/***
向下更新
***/
void pushDown(int rt){
    if(sgt[rt].mark!=0) {
        sgt[rt<<1].mark+=sgt[rt].mark;
        sgt[rt<<1|1].mark+=sgt[rt].mark;
        sgt[rt<<1].sum += (ll)(sgt[rt<<1].r-sgt[rt<<1].l+1)*sgt[rt].mark;//这个地方容易写错，写成sgt[rt<<1].mark，注意一下
        sgt[rt<<1|1].sum +=(ll)(sgt[rt<<1|1].r-sgt[rt<<1|1].l+1)*sgt[rt].mark;

        //sgt[rt<<1].mmax += sgt[rt].mark;
        //sgt[rt<<1|1].mmax +=sgt[rt].mark;

        sgt[rt].mark=0;
    }
}
/***
建树
build(1,n,1)
***/
void build(int l,int r,int rt){
     sgt[rt].l=l;
     sgt[rt].r=r;
     sgt[rt].mark=0;
     if(l==r){
        sgt[rt].sum=a[l];//sum
        //sgt[rt].mmax=a[l];
        return;
     }

     int m=(l+r)>>1;
     build(l,m,rt<<1);build(m+1,r,rt<<1|1);
     pushUp(rt);
}
/***
区间加减
Interval_query(addNumber,L,R,1)
***/
void Interval_update(int add,int l,int r,int rt){

    if(l<=sgt[rt].l && r>=sgt[rt].r){
        sgt[rt].mark += add;
        sgt[rt].sum  += (ll)add*(sgt[rt].r - sgt[rt].l+1);//sum
        //sgt[rt].mmax  += add;
        return;
    }
    pushDown(rt);
    int m = (sgt[rt].l + sgt[rt].r) >> 1;
    if(l <= m) {
        Interval_update(add,l,r,rt<<1);
    }
    if(r > m){
        Interval_update(add,l,r,rt<<1|1);
    }
    pushUp(rt);
}
/***
区间查询
Interval_query(L,R,1)
***/
ll Interval_query(int l,int r,int rt){

    if(l<=sgt[rt].l && r>=sgt[rt].r){
        return sgt[rt].sum;
        //return sgt[rt].mmax;
    }
    pushDown(rt);
    int m = (sgt[rt].l + sgt[rt].r) >> 1;
    ll res = 0;
    //ll mmax = LONG_LONG_MIN;
    if(l <= m) {
        res += Interval_query(l,r,rt<<1);
        //mmax = max(mmax,Interval_query(l,r,rt<<1));
    }
    if(r > m){
        res += Interval_query(l,r,rt<<1|1);
        //mmax = max(mmax,Interval_query(l,r,rt<<1|1));
    }
    return res;
    //return mmax;
}

int main(){
    int n,Q;  
    char str[8];   
    int b,c,d;
    while(~scanf("%d%d",&n,&Q)){
          for(int i=1;i<=n;i++){
             scanf("%d",&a[i]);
          }
          build(1,n,1);
          for(int i=0;i<Q;i++){
               scanf("%s",str);
               if(str[0]=='Q'){
                    scanf("%d%d",&b,&c);
                    printf("%I64d\n",Interval_query(b,c,1));
               }
               else if(str[0]=='C'){
                    scanf("%d%d%d",&b,&c,&d);
                    Interval_update(d,b,c,1);
               }
          }
    }
    return 0;
}

 二、hdu 1698 成段赋值 区间求和（本题求了1~n这个大区间，可以在update里面直接更新，最后输出sgt[1].sum）

#include <cstdio>
#include <iostream>
#include <climits>
#define maxn 100000+1000
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
using namespace std;


typedef long long ll;
struct node{
    int l,r;
    ll sum,mark;
    int mmax;
}sgt[maxn<<2];
void pushUp(int rt){
    sgt[rt].sum = (sgt[rt << 1].sum + sgt[rt<<1|1].sum);
}

void pushDown(int rt){
    if(sgt[rt].mark!=0) {
        sgt[rt<<1].mark=sgt[rt].mark;
        sgt[rt<<1|1].mark=sgt[rt].mark;
        sgt[rt<<1].sum = (ll)(sgt[rt<<1].r-sgt[rt<<1].l+1)*sgt[rt].mark;//这个地方容易写错，写成sgt[rt<<1].mark，注意一下
        sgt[rt<<1|1].sum =(ll)(sgt[rt<<1|1].r-sgt[rt<<1|1].l+1)*sgt[rt].mark;
        
        //sgt[rt<<1].mmax = sgt[rt].mark;  
        //sgt[rt<<1|1].mmax =sgt[rt].mark;  
        sgt[rt].mark=0;
    }
}
/***
建树
build(1,n,1)
***/
void build(int l,int r,int rt){
     sgt[rt].l=l;
     sgt[rt].r=r;
     sgt[rt].mark=0;
     if(l==r){
        sgt[rt].sum=1;//sum
        //sgt[rt].mmax=a[l];
        return;
     }

     int m=(l+r)>>1;
     build(l,m,rt<<1);build(m+1,r,rt<<1|1);
     pushUp(rt);
}
void Interval_set(int val,int l,int r,int rt){

    if(l<=sgt[rt].l && r>=sgt[rt].r){
        sgt[rt].mark = val;
        sgt[rt].sum  = (ll)val*(sgt[rt].r - sgt[rt].l+1);//sum
        //sgt[rt].mmax  = val;
        return;
    }
    pushDown(rt);
    int m = (sgt[rt].l + sgt[rt].r) >> 1;
    if(l <= m) {
        Interval_set(val,l,r,rt<<1);
    }
    if(r > m){
        Interval_set(val,l,r,rt<<1|1);
    }
    pushUp(rt);
}
/***
区间查询
Interval_query(L,R,1)
***/
ll Interval_query(int l,int r,int rt){

    if(l<=sgt[rt].l && r>=sgt[rt].r){
        return sgt[rt].sum;
        //return sgt[rt].mmax;
    }
    pushDown(rt);
    int m = (sgt[rt].l + sgt[rt].r) >> 1;
    ll res = 0;
    //ll mmax = LONG_LONG_MIN;
    if(l <= m) {
        res += Interval_query(l,r,rt<<1);
        //mmax = max(mmax,Interval_query(l,r,rt<<1));
    }
    if(r > m){
        res += Interval_query(l,r,rt<<1|1);
        //mmax = max(mmax,Interval_query(l,r,rt<<1|1));
    }
    return res;
    //return mmax;
}

int main(){
    int t,cas=1;
    scanf("%d",&t);
    while(t--){
        int n,Q;
        scanf("%d%d",&n,&Q);
        build(1,n,1);
        while(Q--){
            int l,r,val;
            scanf("%d%d%d",&l,&r,&val);
            Interval_set(val,l,r,1);
        }
        printf("Case %d: The total value of the hook is %I64d.\n",cas++,Interval_query(1,n,1));

    }
    return 0;
}