引入:
线段树是算法竞赛中常用的用来维护 区间信息 的数据结构。
实现单点修改、区间修改、区间查询(区间求和,求区间最大值,求区间最小值)等操作。
复杂度O(logN)
线段树的建树:
过程:
线段树将每个长度不为1的区间划分成左右两个区间递归求解,把整个线段划分为一个树形结构,通过合并左右两区间信息来求得该区间的信息。
我们一区间最大值为例:(看图一定能明白!)
实现:
struct SegmentTree{
int l,r,dat;
}t[N*4];//struct存储线段树,注意开4倍,这和二叉树的性质有关
void build(int p,int l,int r){//p代表区间l,r
t[p].l=l;t[p].r=r;
if(l==r){
t[p].dat=a[l];
return;
}
int mid=(l+r)/2;
build(p*2,l,mid);//左子节点[l,mid],编号为p*2
build(p*2+1,mid+1,r);//右子节点[mid+1,r],编号为p*2+1
t[p].dat=max(t[p*2].dat,t[p*2+1].dat);//递归取区间最大值
}
build(1,1,n);//调用入口
线段树的单点修改:
把A[x]的值修改为v
过程:
递归查找要修改的点所代表的区间[x,x]的叶节点,然后递归向上更新[x,x]及其所有祖先,以维护区间信息。复杂度O(logN)。如图:(p表示图中最多的数字,后面都是这个意思不在赘述)
主要利用二分思想!
实现:
void change(int p,int x,int v){
if(t[p].l==t[p].r){//找到叶节点
t[p].dat=v;
return;
}
int mid=(t[p].l+t[p].r)/2;
if(x<=mid) change(p*2,x,v);//x在左区间
else change(p*2+1,x,v);//x在右区间
t[p].dat=max(t[p*2].dat,t[p*2+1].dat);//递归向上维护区间信息
}
change(1,x,v);//调用入口
线段树的区间查询:
比如求区间[l,r]的最大值(最小值,区间合。。。)操作。
过程:
我们有区间最大值为例
类比单点修改,就是有一些不一样,详见下图: