线段树模板

一、简单的线段树(区间最值查询,单点修改)

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#define N 100002
using namespace std;
int a[N];
struct Node{
    int l,r;
    int data;
}t[N*4];//注意将数组至少开至最大n的4倍,1+2+4+......2^n+……

void bulid(int p,int l,int r){
    t[p].l=l,t[p].r=r;
    if(l==r){
        t[p].data=a[l];
        return ;
    }
    int mid=(l+r)>>1;
    bulid(p*2,l,mid);
    bulid(p*2+1,mid+1,r);
    t[p].data=max(t[p*2].data,t[p*2+1].data);
}
int query(int p,int l,int r){
    if(l<=t[p].l&&r>=t[p].r){
        return t[p].data;
    }
    int mid=(t[p].l+t[p].r)>>1;
    int val=-(1<<30);
    if(mid>=l) val=max(val,query(p*2,l,r));
    if(mid<r) val=max(val,query(p*2+1,l,r));
    return val;
}
void change(int p,int x,int val){
    if(t[p].l==t[p].r) {
        t[p].data=val;
        return;
    }
    int mid=(t[p].l+t[p].r)>>1;
    if(x<=mid) change(p*2,x,val);
    else  change(p*2+1,x,val);
    t[p].data=max(t[p<<1].data,t[p<<1+1].data);
}
int main()
{
    int n;
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    bulid(1,1,n);
    int tmp,m,x,val,l,r;
    for(int i=1;i<=m;i++){
        cin>>tmp;
        if(tmp==1){
            cin>>x>>val;
            change(1,x,val);
        }
        else {
            cin>>l>>r;
            cout<<query(1,l,r)<<endl;
        }
    }
}

二、线段树+延迟操作
功能:
1、区间同时加上某一个数
2、查询区间和

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100001
#define ll long long
using namespace std;
struct Node
{
    int l,r;
    ll sum,add;//add是sum的延迟操作变量
}t[N*4];
int a[N];
void bulid(int p,int l,int r){
    t[p].l=l,t[p].r=r;
    if(l==r){
        t[p].sum=a[l];
        return ;
    }
    int mid=(l+r)>>1//分成两个部分;
    bulid(p*2,l,mid);
    bulid(p*2+1,mid+1,r);
    t[p].sum=t[p*2].sum+t[p*2+1].sum;
}

void spread(int p){
    if(t[p].add){//将延迟操作向下传递,注意本身设置延迟时,sum没有延迟
        t[p*2].sum +=t[p].add*(t[2*p].r-t[2*p].l+1);
        t[p*2+1].sum+=t[p].add*(t[2*p+1].r-t[2*p+1].l+1);
        t[2*p+1].add+=t[p].add;
        t[2*p].add+=t[p].add;
        t[p].add=0;//清除延迟
    }
}

void change(int p,int l,int r,int d){
    if(l<=t[p].l&&t[p].r<=r){
        t[p].sum+=(ll)d*(t[p].r-t[p].l+1);
        t[p].add+=d;//当可包含当前区间被修改区间包含时,时无需操作该区间
        return;
    }
    spread(p);//判断是否有延迟,有则下传
    int mid=(t[p].l+t[p].r)/2;
    if(l<=mid) change(p*2,l,r,d);
    if(r>mid) change(p*2+1,l,r,d);
    t[p].sum=t[p*2].sum+t[p*2+1].sum;
}

ll  ask(int p,int l,int r){
    if(l<=t[p].l&&t[p].r<=r){
        return t[p].sum;
    }
    spread(p);//如果需要用到的区间存在延迟,下传
    int mid=(t[p].l+t[p].r)>>1;
    ll val=0;
    if(l<=mid) val+=ask(p*2,l,r);
    if(r>mid) val+=ask(p*2+1,l,r);
    return val;
}
int main()
{
    int n,m,tmp,l,r,x,val;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",a+i);
    bulid(1,1,n);
    for(int i=1;i<=m;i++){
        scanf("%d",&tmp);
        if(tmp==2){
            scanf("%d%d",&l,&r);
            printf("%lld\n",ask(1,l,r));
        }
        if(tmp==1){
            scanf("%d%d%d",&l,&r,&val);
            change(1,l,r,val);
        }
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值