栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > C/C++/C#

线段树

C/C++/C# 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

线段树

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

现在我们来看这张图,我们会发现,区间 [ 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[itimes2]+tree[itimes2+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;
}
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/971246.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号