线段树基本操作

17 篇文章 0 订阅
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<string>
#include<vector>
#include<string.h>
#include<map>
#include<cmath>
#include<queue>
#include<stack>
#define ll long long
#define INF 0x7fffffff
#define MAX 0x3f3f3f3f
#define maxn 100005
#define ull unsigned long long
using namespace std;

int a[100005];
//树用结构体表示
struct tree{
    int l,r;
    ll pre,add;//pre 该节点维护的值,add 懒标记
}t[400050];
//建树
void bulid(int p,int l,int r){
    t[p].l=l;//p节点维护的区间为l到r
    t[p].r=r;
    if(l==r){//特殊情况l=r 这个区间就只有一个数
        t[p].pre=a[l];
        return;
    }
    int mid=l+r>>1;
    bulid(p*2,l,mid);//左子树
    bulid(p*2+1,mid+1,r);//右子树
    t[p].pre=t[p*2].pre+t[p*2+1].pre;//维护的值等于左儿子加右儿子
}
//懒标记
void dd(int p){
    if(t[p].add){//如果懒标记不为0,就将其下传,修改左右儿子维护的值
        t[p*2].pre+=t[p].add*(t[p*2].r-t[p*2].l+1);
        t[p*2+1].pre+=t[p].add*(t[p*2+1].r-t[p*2+1].l+1);
        t[p*2].add+=t[p].add;//为左右儿子加上懒标记
        t[p*2+1].add+=t[p].add;
        t[p].add=0;//既已下传,将该节点的懒标记清0
    }
}
//区间修改
void change(int p,int x,int y,int z){
    if(x<=t[p].l && y>=t[p].r){//要修改的区间覆盖了当前节点,就对其进行修改
        t[p].pre+=(ll)z*(t[p].r-t[p].l+1);//注意加一
        t[p].add+=z;//懒标记
        return;
    }
    dd(p);//下传懒标记,继续向下找
    int mid=t[p].l+t[p].r>>1;
    if(x<=mid) change(p*2,x,y,z);//如果要修改的区间覆盖了左儿子,就修改左儿子
    if(y>mid) change(p*2+1,x,y,z);//如果要修改的区间覆盖了右儿子,就修改右儿子
    t[p].pre=t[p*2].pre+t[p*2+1].pre;//最终维护的值等于左儿子的值+右儿子的值
}
//区间查询
ll findd(int p,int x,int y){
    if(x<=t[p].l&&y>=t[p].r)
        return t[p].pre;//如果被覆盖,就返回维护的值
    dd(p);//下传懒标记
    int mid=t[p].l+t[p].r>>1;
    ll ans=0;
    if(x<=mid) ans+=findd(p*2,x,y);//返回左右儿子的和
    if(y>mid) ans+=findd(p*2+1,x,y);
    return ans;
}

int main(){
    int n,m,i,q,x,y,z;
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;i++)
        scanf("%d",&a[i]);
    bulid(1,1,n);//头节点编号是1,维护区间是1到n
    for(i=1;i<=m;i++)
    {
        scanf("%d",&q);
        if(q==1){
            scanf("%d%d%d",&x,&y,&z);
            change(1,x,y,z);//从头节点开始,要修改的区间,区间每个数加z
        }
        else{
            scanf("%d%d",&x,&y);
            cout<<findd(1,x,y)<<endl;//从头节点开始,查询区间各数之和
        }
    }
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

When_3*four

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值