Segment-Tree
Updated:
线段树–动态区间查询
前言:
现在手上有一个数组a[n],让我们来求一个区间的和
这简直太简单了!
我们写一个前缀和就可以在O(1)解决问题了!
问题升级:
有两种操作,一个是修改其中某个变量的值,
另一个是求区间和
这。。我只能O(n²)去做了==
别怕, 线段树可以解决你的问题。
线段树–Segment Tree
既然是线段树,那么一定具有一个树的结构。
这个树和其他树不一样,线段树是用来维护我们手中的数组的
树的结点有这么几个属性:l,r,w
l代表left,是这个结点对应区间的左端点
r代表right,是这个结点对应区间的右端点
w代表这个区间内的区间和
1 | struct node{ |
那我们来想一下大致的过程吧:
首先我们要对于初始的数组a[n]来把这个线段树建立起来
然后,如果要修改一个值,我们就要修改对应叶子结点和它所有的父节点
查找区间和emmm
把一个大区间分成许多我们维护的小区间就好了~
看上去挺简单的
建树–build_tree()
建树也好,建堆也好,这种操作我们经常使用,就是初始化嘛~
我们从根节点开始初始化,把区间逐渐二分下去。
1 | void build_tree(int x,int left,int right){//x是当前节点标号 |
这样我们就能递归的建立一个线段树了。
修改–change()
刚刚的代码里出现了什么奇怪的东西!
是的,我们调用了还没写的change函数
change,就是把一个节点的值进行修改(现在我们简单的定义为加法)
获取的参数是结点的序号,还有改动的值
然后递归向上修改所有的父亲节点:
1 | void change(int x,int a){//x节点的值修改为tree[x].w+a |
查询–find()
那么,我们拿到的指令是,修改a[i]
怎么通过a[i]找到对应的那个叶子结点呢?
二分嘛:
1 | int find(int x,int p,int left,int right){//返回a[x]对应的线段树中的p |
返回区间和–add()
我们已经可以修改并且维护这颗大树了!
接下来只要算算区间和究竟是多少就好了
我们拿到一个区间,这个区间很大可能没有直接与之对应的节点
往往可能是5个,6个节点拼凑出要求的区间
写几个判断语句,然后递归:
1 | int add(int x,int left,int right){//返回区间和 |
调用的时候,x为1,因为要从根节点开始寻找
这样我们就可以进行修改和在线查询了。
前缀和求区间和的方法我们称为离线算法。
而今天的线段树是在线算法。
不同的情况,对于在线离线的要求是不一样的。
当然,能写前缀和我绝对不写线段树
最后祝各位OIer武运昌隆!!!