概念
线段树(Segment Tree)是一种基于分治思想的二叉树结构,用于在区间上进行信息统计。与按照二进制位(2的次幂)进行区间划分的树状数组相比,线段树是一种更加通用的结构:
1、线段树的每个节点都代表一个区间。
2、线段树具有唯一根节点,代表的区间是整个统计范围,如[1,N]。
3、线段树的每个叶节点都代表一个长度为1的元区间[x,x]。
4、对于每个节点[l,r],它的左子节点是[l ,mid],右子节点[mid+1,r],其中mid=(l+r)/2(向下取整)。
线段树结构:
图中展示了一棵线段树。可以发现,除去树的最后一层,整棵线段树一定是一棵满二叉树。树的深度为 i n t ( l o g 2 N ) + 1 int(log2N) + 1 int(log2N)+1( N N N为树上节点数)。因此,我们可以按照与二叉堆类似的“父子2倍”节点编号方法存储。
单点修改,区间查询
实现
1.建树
拿区间最大值来说:因为一个父节点是他两个子节点的并集,所以父节点一定是存储两个子节点的最大值,因此我们可以从上往下递归查找,找到一个节点是叶节点时就将他赋值为序列的这个位置,然后用这个节点来更新它的父节点即可。
void Build_Tree(int p,int l,int r) {
edge[p].l = l;
edge[p].r = r;
if(l == r) {
edge[p].Max_Data = a[l];
return;
}
int mid = (l + r) >> 1;
Build_Tree(p * 2,l,mid);
Build_Tree(p * 2 + 1,mid + 1,r);
edge[p].Max_Data = max(edge[p * 2].Max_Data,edge[p * 2 + 1].Max_Data);
return;
}
2.更新
同样上个例子,因为更新时只更新一个节点,所以就只用查找他一个节点再回溯更新,当查找到的区间包含了这个值,就递归这个区间,就不递归。依然,找到叶节点后,一个一个往上更新至根节点。
void Change_Tree(int l,int r,int x,int y,int p) {
if(l > x || r < x) return;
if(l == r) {
edge[p].Max_Data = y;
return;
}
int mid = (l + r) >> 1;
Change_Tree(l,mid,x,y,p * 2);
Change_Tree(mid + 1,r,x,y,p * 2 + 1);
edge[p].Max_Data = max(edge[p * 2].Max_Data,edge[p * 2 + 1].Max_Data);
}
3.查找
查找时也不难,就和更新时差不多,有交集就查找,如果完全包含就直接更新,如果无交集就结束。然后用最后找到的值在过程中比一下max就行了。
int Query_Tree(int p,int x,int y) {
if(edge[p].l >= x && edge[p].r <= y) return edge[p].Max_Data;
int mid = (edge[p].l + edge[p].r) >> 1,val = -0x3f3f3f3f;
if(mid >= x) val = max(val,Query_Tree(p * 2,x,y));
if(mid < y) val = max(val,Query_Tree(p * 2 + 1,x,y));
return val;
}
题目
//模板代码
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int MAXN = 1e6 + 5;
int n,a[MAXN],q;
struct Segment_Tree{
int l,r,Max_Data;
}edge[MAXN * 4];
void Build_Tree(int p,int l,int r) {
edge[p].l = l;
edge[p].r = r;
if(l == r) {
edge[p].Max_Data = a[l];
return;
}
int mid = (l + r) >>