摘要
线段树是什么??线段树怎么写??
如果你在考提高组前一天还在问这个问题,那么你会与一等奖失之交臂;如果你还在冲击普及组一等奖,那么这篇文章会浪费你人生中宝贵的5~20分钟。
上面两句话显而易见,线段树这个数据结构是一个从萌新到正式OI选手的过渡,是一个非常重要的算法,也是一个对于萌新来说较难的算法。不得不说,我学习了这个算法5遍左右才有勇气写的这篇博客。
但是,对于OI正式选手来说,线段树不是算法,应该是一种工具。她能把一些对于区间(或者线段)的修改、维护,从O(N)的时间复杂度变成O(logN)。
废话不说,这篇博客会分为四部:
第一部:线段树概念引入
第二部:简单(无pushdown)的线段树
第三部:区间+/-修改与查询
第四部:区间乘除修改与查询
总结
受篇幅限制,本次只介绍前两部分。
第一部 概念引入
线段树是一种二叉树,也就是对于一个线段,我们会用一个二叉树来表示。比如说一个长度为4的线段,我们可以表示成这样:
这是什么意思呢?如果你要表示线段的和,那么最上面的根节点的权值表示的是这个线段1~4的和。根的两个儿子分别表示这个线段中1~2的和,与3~4的和。以此类推。
然后我们还可以的到一个性质:节点i的权值=她的左儿子权值+她的右儿子权值。因为1~4的和就是等于1~2的和+2~3的和。
根据这个思路,我们就可以建树了,我们设一个结构体tree,tree[i].l和tree[i].r分别表示这个点代表的线段的左右下标,tree[i].sum表示这个节点表示的线段和。
我们知道,一颗二叉树,她的左儿子和右儿子编号分别是她*2和她*2+1
再根据刚才的性质,得到式子:tree[i].sum=tree[i*2].sum+tree[i*2+1].sum;就可以建一颗线段树了!代码如下:
1
inline void build(int i,int l,int r){//递归建树
tree[i].l=l;tree[i].r=r;
if(l==r){//如果这个节点是叶子节点
tree[i].sum=input[l];
return ;
}
int mid=(l+r)>>1;
build(i*2,l,mid);//分别构造左子树和右子树
build(i*2+1,mid+1,r);
tree[i].sum=tree[i*2].sum+tree[i*2+1].sum;//刚才我们发现的性质return ;
}
嗯,这就是线段树的构建,你可能会问为什么要开好几倍的内存去储存一条线段。这是因为我们还没有让这个过大的数组干一些实事,那么什么是实事呢?让我们进入下一部(在你看懂这一部的情况下)
第二部 简单(无pushdown)的线段树
1、单点修改,区间查询
其实这一章开始才是真正的线段树,我们要用线段树干什么?答案是维护一个线段(或者区间),比如你想求出一个1~100区间中,4~67这些元素的和,你会怎么做?朴素的做法是for(i=4;i<=67;i++) sum+=a[i],这样固然好,但是算得太慢了。
我们想一种新的方法,先想一个比较好画图的数据,比如一个长度为4的区间,分别是1、2、3、4,我们想求出第1~3项的和。按照上一部说的,我们要建出一颗线段树,其中点权(也就是红色)表示和:
然后我们要求1~3的和,我们先从根节点开始查询,发现她的左儿子1~2这个区间和答案区间1~3有交集,那么我们跑到左儿子这个区间。
然后,我们发现这个区间1~2被完全包括在答案区间1~3这个区间里面,那就把她的值3返回。
我们回到了1~4区间,发现她的右儿子3~4区间和答案区间1~3有交集,那么我们走到3~4区间
到了3~4区间,我们发现她并没有完全包含在答案区间1~3里面,但发现她的左儿子3~3区间和1~3区间又交集,那么久走到3~3区间
到了3~3区间,发现其被答案区间完全包含,就返回她的值3一直到最开始
3~3区间的3+1~2区间的3=6,我们知道了1~3区间和为6.
有人可能会说你这样是不是疯了,我那脚都能算出1+2+3=6,为什么这么麻烦?!
因为这才几个数,如果一百万个数,这样时间会大大增快。
我们总结一下,线段树的查询方法:
1、如果这个区间被完全包括在目标区间里面,直接返回这个区间的值
2、如果这个区间的左儿子和目标区间有交集,那么搜索左儿子
3、如果这个区间的右儿子和目标区间有交集,那么搜索右儿子
写成代码,就会变成这样:
1
inline int search(int i,int l,int r){
if(tree[i].l>=l && tree[i].r<=r)//如果这个区间被完全包括在目标区间里面,直接返回这个区间的值
return tree[i].sum;
if(tree[i].rr) return 0;//如果这个区间和目标区间毫不相干,返回0
int s=0;
if(tree[i*2].r>=l) s+=search(i*2,l,r);//如果这个区间的左儿子和目标区间又交集,那么搜索左儿子
if(tree[i*2+1].l<=r) s+=search(i*2+1,l,r);//如果这个区间的右儿子和目标区间又交集,那么搜索右儿子
return s;
}
关于那几个if的条件一定要看清楚,最好背下来,以防考场上脑抽推错。
然后,我们怎么修改这个区间的单点,其实这个相对简单很多,你要把区间的第dis位加上k。
那么你从根节点开始,看这个dis是在左儿子还是在右儿子,在哪往哪跑,
然后返回的时候,还是按照tree[i].sum=tree[i*2].sum+tree[i*2+1].sum的原则,更新所有路过的点
如果不理解,我还是画个图吧,其中深蓝色是去的路径,浅蓝色是返回的路径,回来时候红色的+标记就是把这个点加上这个值。
把这个过程变成代码,就是这个样子:
1
inline void add(int i,int dis,int k){
if(tree[i].l==tree[i].r){//如果是叶子节点,那么说明找到了
tree[i].sum+=k;
return ;
}
if(dis<=tree[i*2].r) add(i*2,dis,k);//在哪往哪跑
else add(i*2+1,dis,k);
tree[i].sum=tree[i*2].sum+tree[i*2+1].sum;//返回更新
return ;
}
2、区间修改,单点查询
区间修改和单点查询,我们的思路就变为:如果把这个区间加上k,相当于把这个区间涂上一个k的标记,然后单点查询的时候,就从上跑道下,把沿路的标记加起来就好。
这里面给区间贴标记的方式与上面的区间查找类似,原则还是那三条,只不过第一条:如果这个区间被完全包括在目标区间里面,直接返回这个区间的值变为了如果这个区间如果这个区间被完全包括在目标区间里面,讲这个区间标记k
具体做法很像,这里贴上代码:1
inline void add(int i,int l,int r,int k){
if(tree[i].l>=l && tree[i].r<=r){//如果这个区间被完全包括在目标区间里面,讲这个区间标记k
tree[i].sum+=k;
return ;
}
if(tree[i*2].r>=l)
add(i*2,l,r,k);
if(tree[i*2+1].l<=r)
add(i*2+1,l,r,k);
}
然后就是单点查询了,这个更好理解了,就是dis在哪往哪跑,把路径上所有的标价加上就好了:
1
void search(int i,int dis){
ans+=tree[i].num;//一路加起来
if(tree[i].l==tree[i].r)
return ;
if(dis<=tree[i*2].r)
search(i*2,dis);
if(dis>=tree[i*2+1].l)
search(i*2+1,dis);
}
不知不觉,这第二章已经结束。这样的简单(原谅我用这个词)线段树,还可除了求和,还可以求区间最小最大值,还可以区间染色。
但是!这样的线段树展现不出来她的魅力,因为区间求和,树状数组比她少了一个很大的常熟。二区间最值,ST的那神乎其技的O(n)查询也能完爆她。这是为什么?因为线段树的魅力还没有展现出来,她最美丽的地方:pushdown还未展现于世,如果你已经对这一章充足的了解,并且能不看博客把洛谷上树状数组模板1、2都能写出来,那么请你进入下一部。
本文作者: Dijkstra·Liu ,原文链接:https://www.cnblogs.com/jason2003/p/9676729.html
信息学直播课程通知
往期精选内容
(点击标题即可查看)
CCF将举办NOI Online能力测试,每组限额报名3万人
NOI Online测试有关事项问答
关于延期举行NOI2020冬令营的通知
CCF关于恢复NOIP竞赛的公告
强基计划解读,一切向着好的方向发展!
CCF关于NOI省队选拔相关问题的补充通知
教育部发布“强基计划”,2020年自主招生不再组织
CSP-J/S认证第一轮初赛S、J组答案
逾10万人参加首次CSP-J/S认证,附CSP-S/J组初赛试题及参考答案
CSP-J/S 各省负责人联系方式!
NOIP2015-2018普及组&提高组初赛试题答案&解析
CCF发布NOI2020省队选拔规定
学好信竞-浅谈信息学竞赛考场策略及程序测试
详细盘点清华姚班 智班,北大 浙大图灵班等多所高校AI专业实力!
大牛云集!清华大学2019年姚班及智班第一届AI本科生名单公布!
清北学堂C++入门公益课视频已上传!
清华大学计算机姚班、人工智能学堂班(智班)选拔方案出炉!
2019年全国69所高校保送生拟录取名单公布,清华北大共349人!
CTS2019 获奖名单分析-信息学强省强校排行榜!
APIO2019中国区获奖名单公布,学军中学 杭州二中等表现出色
五大学科竞赛国家队23人大名单出炉(各省,中学分布统计)
IOI2019(国际信息学奥林匹克)中国队名单出炉
最详细解析低分进名校三大途径:自主招生、综合评价、高校专项计划!
2019年保送生资格名单公示
再见,OI-大牛HZW亲笔,分享OI生涯记录,不变的是坚持和热爱!
NOIP复赛知识点简述及复赛算法总结!
根据信息学竞赛之路带你了解信息学竞赛流程
全国自主招生高校对各项学科竞赛奖项要求大汇总!签约路径
一个清华保送生妈妈对竞赛的感受,自主招生家长都要看看!
为什么这些孩子初中就能被清华北大签约
(1) 为什么有“编程思维”和数学能力强的人更优秀?(2)清北独家录制NOIP成功者说学习视频!!!
(3)我们为什么要对孩子进行编程教育?
(4)信息学竞赛答家长问题
1.信息学竞赛,你想了解的知识都在这里
2.信息学奥赛(NOIP)初赛学习方法推荐
3.信息学奥赛(NOIP)复赛学习方法推荐
4.大牛为你推荐十本最适合信息学竞赛的书籍
5.信息学奥赛有那么重要吗?
6.参加编程竞赛对实际工作的用处
7.清北学堂独家录制NOIP考试技巧讲座
8.在线编程挑战赛第一名:我是这么学算法的
9.信息学竞赛如何学习及准备攻略!
10.凭什么我得了信息学奥赛国家一等奖
11.榜样 | 北大降200分要这个诸暨天才少年
12.OI金牌教练胡芳:爱和成长的故事
13.信息学竞赛,一个让孩子不需要再去挤独木桥的方向
14.新学期必须了解的学科竞赛与自主招生时间!
15.北大录取生陈代超:在信息学中找到“思维图谱”
16.国务院发文支持编程教育进入中小学,中国人工智能厚积薄发
(1)NOIP报名,申诉,查成绩方式介绍
(2)NOIP复赛考前需要注意的那些事儿!!!
(3)NOIP测评环境,数据提交你都了解吗?请注意这些问题!
(4)关于NOI系列赛编程语言使用限制的规定
关注「信息学竞赛」
看更多信息学趣闻与知识
↓↓↓