(线段树)模板

来源:互联网 发布:java web 攻击 2 编辑:程序博客网 时间:2024/05/21 03:57

支持的操作有:
区间最大最小值
区间和

# include <iostream># include <string.h>#define maxn 100007  //元素总个数#define ls l,m,rt<<1#define rs m+1,r,rt<<1|1using namespace std;typedef struct shu{    int sum;    int mi;    int mx;}Shu;Shu xianduan[10000];int Sum[maxn<<2],Add[maxn<<2];//Sum求和,Add为懒惰标记int A[maxn],n;//存原数组数据下标[1,n]//PushUp函数更新节点信息 ,这里是求和void PushUp(int rt){    xianduan[rt].sum=xianduan[rt<<1].sum+xianduan[rt<<1|1].sum;    xianduan[rt].mx = max(xianduan[rt].mx,xianduan[rt].mx);    xianduan[rt].mi = min(xianduan[rt].mi,xianduan[rt].mi);}//Build函数建树void Build(int l,int r,int rt){ //l,r表示当前节点区间,rt表示当前节点编号    if(l==r) {//若到达叶节点       xianduan[rt].sum=A[l];//储存数组值       xianduan[rt].mx=A[l];       xianduan[rt].mi=A[l];        return;    }    int m=(l+r)>>1;    //左右递归    Build(l,m,rt<<1);    Build(m+1,r,rt<<1|1);    //更新信息    PushUp(rt);}//单点更新void Update(int L,int C,int l,int r,int rt){    //l,r表示当前节点区间,rt表示当前节点编号    //L 代表更新的节点的位置,C为加上或减去的值    if(l==r){//到叶节点,修改         xianduan[rt].sum+=C;         xianduan[rt].mx+=C;         xianduan[rt].mi+=C;         Add[rt] += C;        return;    }    int m=(l+r)>>1;    //根据条件判断往左子树调用还是往右    if(L <= m) Update(L,C,l,m,rt<<1);    else       Update(L,C,m+1,r,rt<<1|1);    PushUp(rt);//子节点更新了,所以本节点也需要更新信息}//延迟标志void PushDown(int rt,int ln,int rn){    //ln,rn为左子树,右子树的数字数量。    if(Add[rt]){        //下推标记        Add[rt<<1]+=Add[rt];        Add[rt<<1|1]+=Add[rt];        //修改子节点的Sum使之与对应的Add相对应       xianduan[rt<<1].sum+=Add[rt]*ln;       xianduan[rt<<1|1].sum+=Add[rt]*rn;        xianduan[rt<<1].mx+=Add[rt]*ln;       xianduan[rt<<1|1].mx+=Add[rt]*rn;        xianduan[rt<<1].mi+=Add[rt]*ln;       xianduan[rt<<1|1].mi+=Add[rt]*rn;        //清除本节点标记        Add[rt]=0;    }}//区间更新void Update1(int L,int R,int C,int l,int r,int rt){    //L,R表示操作区间,l,r表示当前节点区间,rt表示当前节点编号    if(L <= l && r <= R)    {//如果本区间完全在操作区间[L,R]以内        xianduan[rt].sum+=C*(r-l+1);//更新数字和,向上保持正确        xianduan[rt].mx+=C*(r-l+1);        xianduan[rt].mi+=C*(r-l+1);        Add[rt]+=C;//增加Add标记,表示本区间的Sum正确,子区间的Sum仍需要根据Add的值来调整        return ;    }    int m=(l+r)>>1;    PushDown(rt,m-l+1,r-m);//下推标记    //这里判断左右子树跟[L,R]有无交集,有交集才递归    if(L <= m) Update1(L,R,C,l,m,rt<<1);    if(R >  m) Update1(L,R,C,m+1,r,rt<<1|1);    PushUp(rt);//更新本节点信息}//求区间和int Query(int L,int R,int l,int r,int rt){//L,R表示操作区间,l,r表示当前节点区间,rt表示当前节点编号    if(L <= l && r <= R){        //在区间内,直接返回        return xianduan[rt].sum;    }    int m=(l+r)>>1;    //下推标记,否则Sum可能不正确    PushDown(rt,m-l+1,r-m);    //累计答案    int ANS=0;    if(L <= m) ANS+=Query(L,R,l,m,rt<<1);    if(R >  m) ANS+=Query(L,R,m+1,r,rt<<1|1);    return ANS;}//求区间最小值int Qmin(int L,int R,int l,int r,int rt){//L,R表示操作区间,l,r表示当前节点区间,rt表示当前节点编号    if(L>r||R<l)        return 100000;    if(L <= l && r <= R)    {        //在区间内,直接返回        return xianduan[rt].mi;    }    int m=(l+r)>>1;    //下推标记,否则Sum可能不正确    PushDown(rt,m-l+1,r-m);    //累计答案  return   min(Query(L,R,l,m,rt<<1),Query(L,R,m+1,r,rt<<1|1));}//求区间最大值int Qmax(int L,int R,int l,int r,int rt){//L,R表示操作区间,l,r表示当前节点区间,rt表示当前节点编号    if(L>r||R<l)        return 100000;    if(L <= l && r <= R)    {        //在区间内,直接返回        return xianduan[rt].mx;    }    int m=(l+r)>>1;    //下推标记,否则Sum可能不正确    PushDown(rt,m-l+1,r-m);    //累计答案  return   max(Query(L,R,l,m,rt<<1),Query(L,R,m+1,r,rt<<1|1));}int main(void){    A[1] = 1;    A[2] = 2;    A[3] = 3;    A[4] = 4;    n = 4;    //建树    Build(1,n,1);   // 点修改  /* Update(1,-1,1,n,1);   Update(2,-1,1,n,1);   Update(3,-1,1,n,1);   Update(4,-1,1,n,1);*/    //区间修改   Update1(1,4,1,1,n,1);   // 区间查询.    int ANS=Qmax(2,3,1,n,1);    cout <<ANS <<endl;    return 0;}
原创粉丝点击