线段树是什么?简而言之,这就是一种算法,或者说是一种工具,是一种将区间修改、查询最大值、最小值、和、积等等的 O ( n ) O(n) O(n) 方法降为 O ( l o g ( n ) ) O(log(n)) O(log(n))的做法
1.线段树简介
加入现在你有一个数组 a a a 现在有 1,2,3,4 四个数,那么线段树维护这个数组的和将会是这样的:
可见,线段树其实就是一个二分的结构。那它是如何操作的呢?
2 线段树的操作
2.1结构
struct tree{
int l ,r;//左端点和右端点
long long sum,/*要维护的数据*/ lazy;//下传标志
}a[400005];//要是数据大小的4倍
2.2建树
建树的操作简而言之就是将一个数组拆成一个一个的结构,每一个父亲节点的值都是它左儿子和右儿子的值的操作,所以建树的操作就相当于一个递归。(具体过程类似于树状数组)
void building (int x, int l, int r)//建树
{
a[x].l = l;
a[x].r = r;//左右端点
if(a[x].l == a[x].r) {a[x].sum = f[l]; return;}
long long mid = (l + r) / 2;
building(x * 2/*左子树的编号是当前的两倍*/ ,l, mid);//下一个右端点
building(x * 2 + 1,mid + 1, r); //同理
a[x].sum = a[x * 2].sum + a[x * 2 + 1].sum;//求要维护的值
return ;
}
2.3修改
将一定的数据下放,沿着线段树的二分结构不断修改。
void change(int x,int l,int r,int y)
{
if (a[x].l>=l&&a[x].r<=r){
a[x].sum+=(a[x].r-a[x].l+1)*y;
a[x].lazy+=y;
return;
}//修改的区间完全覆盖,直接修改
if (a[x].r<l||a[x].l>r) return;//修改的区间完全不覆盖
jiafa(2*x,l,r,y);
jiafa(2*x+1,l,r,y);//部分覆盖,向下一个区间探索
a[x].sum=a[2*x].sum+a[2*x+1].sum;//求当前的值
return;
}
2.4查询
详见注释
long long find(int x, int l, int r)
{
if(a[x].l >= l && a[x].r <= r){return a[x].sum;}//完全覆盖,返回当前区间需维护的值
if(a[x].r < l || a[x].l > r){return 0;}//完全不覆盖,返回伪值
if(a[x].lazy) xiangxia(x);//下放标记
return find(x * 2, l, r) + find (x * 2 + 1, l ,r);//继续向下查询
}
3 线段树之一个tag标志
在上文的 2.3 中,读者或许都发现了这样一个变量 l a z y lazy lazy,这个变量是做什么用的呢?
众所周知,在线段树的区间修改中,常常会有一个区间会被多次修改所覆盖,如果说每一次修改后都将数据全部传递到叶子结点,那么就会十分浪费时间可能会直接退化成
O
(
n
)
O(n)
O(n) ,在一些恶心的题目中如果不用
l
a
z
y
lazy
lazy标志,就会
T
L
E
TLE
TLE。实在是可恶至极。
那我们该怎么办呢?我们可以将一定的数据先存进 l a z y lazy lazy 标志中,等到我们需要查询(或者在区间修改中需要提前下放)时在一起传递到子节点。
这个标志的实现的方法也十分简单,如下:
3.1创造lazy标志(以P3372 【模板】线段树 1为例)
详见注释:
void change(int x,int l,int r,int y)
{
if (a[x].l>=l&&a[x].r<=r){
a[x].sum+=(a[x].r-a[x].l+1)*y;
a[x].lazy+=y; //当区间完全覆盖时,将lazy标志存起来
return;
}
if (a[x].r<l||a[x].l>r) return;
jiafa(2*x,l,r,y);
jiafa(2*x+1,l,r,y);
if (a[x].lazy) xiangxia(x);//当区间没有完全覆盖时,要先下放lazy标志到下一层,这样才能正确维护数据
a[x].sum=a[2*x].sum+a[2*x+1].sum;
return;
}
3.2 lazy标志的维护
说完了标志的创建,现在再来说一说如何下放标志(这是标志的维护,也是整个代码中最容易错的地方,也是调试最久的地方)
void xiangxia(int x){
int m = (a[x].l + a[x].r) / 2;
a[2 * x].sum += (m - a[x].l + 1) * a[x].lazy;//下放标记
a[2 * x + 1].sum += (a[x].r - m) * a[x].lazy;
a[2 * x].lazy += a[x].lazy;
a[2 * x + 1].lazy += a[x].lazy;//将子节点的标记叠加,不用继续向下传递
a[x].lazy = 0;//清空标记
return ;
}