线段树

线段树是一种二叉搜索树,首先它满足二叉树,每个结点最多有两个子树,其次,我们知道线段树的每个结点都存储了一段区间,也可以理解为线段,搜索就是在这些线段上进行搜索操作找到我们想要的答案。

在这里插入图片描述

现在我们来看这张图,我们会发现,区间 [ 1 , 13 ] [1,13] [1,13]的和=区间 [ 1 , 7 ] [1,7] [1,7]的和+区间 [ 8 , 13 ] [8,13] [8,13]的和。以此类推,我们可以总结出一个性质:结点 i i i的权值=它的左儿子的权值+它的右儿子权值,即得到公式 t r e e [ i ] = t r e e [ i × 2 ] + t r e e [ i × 2 + 1 ] tree[i]=tree[i\times2]+tree[i\times2+1] tree[i]=tree[i×2]+tree[i×2+1]。建树

struct Tree{//存储val维护结点的值,l,r维护区间范围,add涉及一个懒标记
    int l, r;
    ll val, add;
}seg[N << 2 + 2];

void BuildTree(int rt, int l, int r){//建树
    seg[rt].l = l;
    seg[rt].r = r;
    if(l == r){
        cin >> seg[rt].val;
        return;
    }
    int mid = l + (r - l) >> 1;
    BuildTree(rt << 1, l, mid);
    BuildTree(rt << 1 | 1, mid + 1, r);
    seg[rt].val = seg[rt << 1].val + seg[rt << 1 | 1].val;
}

懒标记:精髓就是打标记和下传操作,由于我们要做的操作时区间加上一个数,所以我们就在区间进行修改时给该区间打上一个懒标记,这样我们就不用在修改它的左右儿子所维护的区间,等到要使用该结点索要维护的值时,再将懒标记下传即可,这样我们就可以节省很多时间,对于每次区间修改和查询,将懒标记下传,就可以节省很多时间。

void spread(int rt){//懒标记
    if(seg[rt].add){//如果懒标记不为0,就将其传下去,并维护左右儿子的值
        seg[rt << 1].val = seg[rt].add * (seg[rt >> 1].r - seg[rt >> 1].l + 1);
        seg[rt << 1 | 1].val = seg[rt].add * (seg[rt << 1 | 1].r - seg[rt << 1 | 1].l + 1);
        seg[rt << 1].add += seg[rt].add;
        seg[rt << 1 | 1].add += seg[rt].add;
        seg[rt].add = 0;//传下去之后将该结点的懒标记清零
    }
}

区间修改

考虑到,我们是将一个区间加上一个数,我们可以从根结点开始不断向下查找,当我们要修改的区间覆盖当前结点覆盖时,我们就把这个区间进行修改并打上懒标记,否则就将懒标记下传,继续进行查找。

void chang(int rt, int l, int r, int k){//修改
    if(seg[rt].l >= l && seg[rt].r <= r){//如果区间被覆盖,就对其进行修改
        seg[rt].val += k * (seg[rt].r - seg[rt].l + 1);
        seg[rt].add += k;//打上懒标记
        return;
    }
    spread(rt);//如果没有被覆盖,既需要哦继续向下查找,考虑儿子所需要维护的区间可能因为上面的懒标记而没有修改
    int mid = (seg[rt].r + seg[rt].l) >> 1;
    if(l <= mid) chang(rt << 1, l, r, k);//如果区间覆盖左儿子,及修改左儿子
    if(r > mid) chang(rt << 1 | 1, l, r, k);//右儿子同理
    seg[rt].val = seg[rt << 1].val + seg[rt << 1 | 1].val;//最终维护的值=做儿子的值+右儿子的值
}

区间查询

查询和修改的操作几乎一样,依旧是从根结点开始查找,当我们要查找的区间被覆盖时,就返回维护的值,否则就将懒标记下传,继续进行查找。

int ask(int rt, int l, int r){//查找
    if(seg[rt].l >= l && seg[rt].r <= r)return seg[rt].val;//如果期间被覆盖,就直接返回其值
    spread(rt);//下传懒标记,并查询左右儿子
    int mid = seg[rt].r + seg[rt].l >> 1;
    ll res = 0;
    if(l <= mid) res += ask(rt << 1, l, r);
    if(r > mid) res += ask(rt << 1 | 1, l, r);//累加答案,返回左右儿子的和
    return res;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

星*湖

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

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

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

打赏作者

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

抵扣说明:

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

余额充值