线段树模板+详细注释

例题

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
#define ll long long
#define MT(a,b) memset(a,b,sizeof(a))
const int maxn=1E5+5;
const int ONF=-0x3f3f3f3f;
const int INF=0x3f3f3f3f;
int num[maxn];
struct node
{
    ll sum,lazy;
}qwe[maxn<<2];//用等比数列的知识可得空间趋近于4N
void update(int root)
{
    qwe[root].sum=qwe[root<<1].sum+qwe[root<<1|1].sum;//每个结点的sum=左儿子+右儿子
}
void build_tree(int l,int r,int root)//建立线段树
{
    qwe[root].lazy=0;//初始化lazy数组标记
    if (l==r){
        qwe[root].sum=num[l];//如果l=r说明此时区间只有一个数,是叶子结点
        return;
    }
    int mid=(l+r)>>1;
    build_tree(l,mid,root<<1);//这是左儿子
    build_tree(mid+1,r,root<<1|1);//这是右儿子
    update(root);//更新这个结点的sum
}
void push_down(int root,int l,int r)//将一个结点的lazy赋给它的儿子们,并且将这个lazy加到它的值上
{
    if (qwe[root].lazy==0)//如果lazy=0,说明不需要处理
        return ;
    int mid=(l+r)>>1;//可能有多次lazy的改变所以选择+=
    qwe[root<<1].lazy+=qwe[root].lazy;//左结点
    qwe[root<<1|1].lazy+=qwe[root].lazy;//右结点
    qwe[root<<1].sum+=qwe[root].lazy*(mid-l+1);
    qwe[root<<1|1].sum+=qwe[root].lazy*(r-mid);
    qwe[root].lazy=0;//初始化这个结点的lazy,因为以及给了儿子并且这个结点的sum已经被lazy改变了
}
void change(int l,int r,int ql,int qr,int root,int val)//让[ql,qr]所有元素加上val
{
    if (l==ql&&r==qr){
        qwe[root].sum+=val*(r-l+1);//如果已经是目标区间了,处理sum
        qwe[root].lazy+=val;//标记上lazy,这样他的子子孙孙也相当于有了保险
        return;
    }
    push_down(root,l,r);//处理lazy标记
    int mid=(l+r)>>1;
    if (qr<=mid) change(l,mid,ql,qr,root<<1,val);//如果目标区间在这个结点的左侧
    else if (ql>mid) change(mid+1,r,ql,qr,root<<1|1,val);//如果目标区间在这个结点的右侧
    else{//如果目标区间既有结点左侧,也存在于结点右侧
        change(l,mid,ql,mid,root<<1,val);//那么相当于在左儿子找[ql,(l+r)/2]的值+在右儿子找[(l+r)/2+1,qr]的值
        change(mid+1,r,mid+1,qr,root<<1|1,val);
    }
    update(root);
}
ll query (int l,int r,int ql,int qr,int root)//查询区间[ql,qr]的总和
{
    if (ql==l&&qr==r)
        return qwe[root].sum;
    push_down(root,l,r);
    int mid=(l+r)>>1;
    if (qr<=mid) return query(l,mid,ql,qr,root<<1);
    else if (ql>mid) return query(mid+1,r,ql,qr,root<<1|1);
    else return query(l,mid,ql,mid,root<<1)+query(mid+1,r,mid+1,qr,root<<1|1);
}
int main ()
{
    char s[2];
    ll tmp,a,b,c;
    int n,q;while (~scanf ("%d%d",&n,&q)){
        for (int i=1;i<=n;i++){
            scanf ("%lld",&tmp);
            num[i]=tmp;
        }
        build_tree(1,n,1);
        for (int i=1;i<=q;i++){
            scanf("%s",s);
            if (s[0]=='C'){
                scanf ("%lld%lld%lld",&a,&b,&c);
                change(1,n,a,b,1,c);
            } else if (s[0]=='Q'){
                scanf ("%lld%lld",&a,&b);
                tmp=query(1,n,a,b,1);
                printf("%lld\n",tmp);
            }
        }
//        for (int i=1;i<=n;i++)
//            printf("%d ",query(1,n,i,i,1));
    }
    return 0;
}

你可能感兴趣的:(线段树模板+详细注释)