模板：线段树

最新推荐文章于 2024-07-20 21:43:30 发布

Coder_Num233

最新推荐文章于 2024-07-20 21:43:30 发布

阅读量250

点赞数 1

文章标签：博客

本文链接：https://blog.csdn.net/Coder_Number233/article/details/53510046

版权

数据结构专栏收录该内容

1 篇文章 0 订阅

订阅专栏

线段树笔记：

函数名称	函数作用
build(l,r,k)	建造线段树
findmx(k,l,r)	查找指定区间内最大值
findmn(k,l,r)	查找指定区间内最小值
countsum(k,l,r)	计算指定区间内元素和
…	……

变量名称	变量意义
tree[i].left	树上节点[i]的左顶点
tree[i].right	树上节点[i]的右顶点
tree[i].value	树上节点[i]所指代线段的区间和
tree[i].max	树上节点[i]所指代线段的区间最大值
tree[i].min	树上节点[i]所指代线段的区间最小值
tree[i].change	未添加树上节点[i]所指代线段的区间是否被修改过
(In function) l	当前所要查找线段的左顶点
(In function) r	当前所要查找线段的右顶点
(In function) mid	当前所要查找线段的中点(递归转移用)
…	……

原理

函数名称	原理解释
build	构建一个区间长度为n的线段树.判断l是否等于r,不等则递归建造左右子树,并处理区间和
findmx	查询区间[l..r]中的最大值.判断两区间是否相交,递归查找子区间内最大值
findmn	查询区间[l..r]中的最小值.判断两区间是否相交,递归查找子区间内最大值
countsum	计算区间[l..r]的和.判断当前区间是否被包含,否则递归查找至包含,最后相加
…	……

代码：

#include<cstdio>
#define N 200001
using namespace std;
struct point{
    int left,right,value,max,min,change;
}t[N];
int num[N>>1];
int n,l,r;
int max(int i,int j){
    return i<j?j:i;
}
int min(int i,int j){
    return i>j?j:i;
}
void build(int l,int r,int k){int mid=(l+r)>>1;
    t[k].left=l,t[k].right=r;
    if(l!=r){
        build(l,mid,k<<1);
        build(mid+1,r,(k<<1)|1);
        t[k].value=t[k<<1].value+t[(k<<1)|1].value;
        t[k].min=t[k<<1].min<t[(k<<1)|1].min?t[k<<1].min:t[(k<<1)|1].min;
        t[k].max=t[k<<1].max>t[(k<<1)|1].max?t[k<<1].max:t[(k<<1)|1].max;
    }else{
        t[k].value=num[l],t[k].min=t[k].value,t[k].max=t[k].value;
    }return;
}
int findmx(int k,int l,int r){
    if(l<1||r>n)return -1;
    int mid=(t[k].left+t[k].right)>>1;
    if(l==t[k].left){
        if(r<t[k].right)
            return r>mid?max(t[k<<1].max,findmx((k<<1)|1,mid+1,r)):findmx(k<<1,l,r);
        return t[k].max;
    }else{
        if(l<=mid)
            return r>mid?max(findmx(k<<1,l,mid),findmx((k<<1)|1,mid+1,r)):findmx(k<<1,l,r);
        return findmx((k<<1)|1,l,r);
    }
}
int findmn(int k,int l,int r){
    if(l<1||r>n)return -1;
    int mid=(t[k].left+t[k].right)>>1;
    if(l==t[k].left){
        if(r<t[k].right)
            return r>mid?min(t[k<<1].min,findmn((k<<1)|1,mid+1,r)):findmn(k<<1,l,r);
        return t[k].min;
    }else{
        if(l<=mid)
            return r>mid?min(findmn(k<<1,l,mid),findmn((k<<1)|1,mid+1,r)):findmn(k<<1,l,r);
        return findmn((k<<1)|1,l,r);
    }
}
int countsum(int k,int l,int r){
    if(l<1||r>n)return -1;
    int mid=(t[k].left+t[k].right)>>1;
    if(l==t[k].left){
        if(r<t[k].right)
            return r>mid?t[k<<1].value+countsum((k<<1)|1,mid+1,r):countsum(k<<1,l,r);
        return t[k].value;
    }else{
        if(l<=mid)
            return r>mid?countsum(k<<1,l,mid)+countsum((k<<1)|1,mid+1,r):countsum(k<<1,l,r);
        return countsum((k<<1)|1,l,r);
    }
}
int main(){int i,j,k;
    scanf("%d",&n);
    for(i=1;i<=n;i++)
        scanf("%d",&num[i]);
    build(1,n,1);
    scanf("%d%d",&l,&r);
    printf("max:%d\nmin:%d\nsum:%d\n",findmx(1,l,r),findmn(1,l,r),countsum(1,l,r));
}

New

支持区间修改、区间求和（这才是线段树优势所在）
代码：

#include<cstdio>
#define N 200001
using namespace std;
struct point{
    int l,r,v,a;
}t[N];
int n,m,k,ans;
void pushup(int x){int tmp=x<<1;
    t[x].v=t[tmp].v+t[tmp+1].v;
}void pushdown(int x){int tmp=x<<1;
    t[tmp].a+=t[x].a,t[tmp+1].a+=t[x].a;
    t[tmp].v+=t[x].a*(t[tmp].r-t[tmp].l+1),t[++tmp].v+=t[x].a*(t[tmp].r-t[tmp].l+1);
    t[x].a=0;
}void build(int x,int l,int r){
    t[x].l=l;
    t[x].r=r;
    t[x].a=0;
    if(l==r){
        scanf("%d",&t[x].v);
        return;
    }int mid=(l+r)>>1,tmp=x<<1;
    build(tmp,l,mid),build(tmp+1,mid+1,r);
    pushup(x);
}void update(int x,int l,int r,int a){
    int tmp=x<<1;
    if(r<t[x].l||l>t[x].r)return;
    if(l<=t[x].l&&r>=t[x].r){
        t[x].a+=a;
        t[x].v+=a*(t[x].r-t[x].l+1);return;
    }if(t[x].a)pushdown(x);
    update(tmp,l,r,a);
    update(tmp+1,l,r,a);
    pushup(x);
}void query(int x,int l,int r){
    if(r<t[x].l||l>t[x].r)return;
    if(l<=t[x].l&&r>=t[x].r){
        ans+=t[x].v;
        return;
    }
    if(t[x].a)pushdown(x);
    int tmp=x<<1,mid=(t[x].l+t[x].r)>>1;
    if(r<=mid)query(tmp,l,r);
    else if(l>mid)query(tmp+1,l,r);
         else query(tmp,l,mid),query(tmp+1,mid+1,r);
}
main(){int i,j,l,r;
    scanf("%d%d",&n,&m);
    build(1,1,n);
    for(i=1;i<=m;i++){
        scanf("%d",&j);
        if(j==1){
            scanf("%d%d%d",&l,&r,&k);
            update(1,l,r,k);
        }else{
            scanf("%d%d",&l,&r);ans=0;
            query(1,l,r);
            printf("%d\n",ans);
        }
    }
}