[简介]
线段树是一种二叉搜索树,与区间树相似,它将一个区间划分成一些单元区间,每个单元区间对应线段树中的一个叶结点。
使用线段树可以快速的查找某一个节点在若干条线段中出现的次数,时间复杂度为
O(logN)
。而未优化的空间复杂度为2N,因此有时需要离散化让空间压缩。
线段树的数据结构可根据题目的需要而进行设计,并没有一个统一的标准,其中一种较为常见的是由于二叉树的父节点编号n与左儿子编号m1、右儿子编号m2之间存在着数量关系: m1=n∗2,m2=n∗2+1 ,因此可以将树的所有节点保存在一维数组中。下面是线段树的核心函数:
#define MAX 50050
int sum[MAX<<2];
int maxn[MAX<<2];
void pushUp(int rt){ //向上更新父节点的值
sum[rt] = sum[rt*2] + sum[rt*2+1]; //若求区间和
maxn[rt] = max(maxn[rt*2], maxn[rt*2+1]); //若求区间最大值
}
void build(int l, int r, int rt){ //树的创建
if(l == r){
scanf("%d", &sum[rt]);
return;
}
int m = (l+r)/2;
build(l,m,rt*2);
build(m+1,r,rt*2+1);
pushUp(rt);
}
void update(int p, int add, int l, int r, int rt){ //叶节点值发生变化时,树的更新
if(l == r){
sum[rt] += add;
return;
}
int m = (l+r)/2;
if(p<=m)
update(p,add,l,m,rt*2);
else
update(p,add,m+1,r,rt*2+1);
pushUp(rt);
}
int getSum(int L, int R, int l, int r, int rt){ //区间和
int ret = 0;
if(L<=l && r<=R)
return sum[rt];
int m = (l+r)/2;
if(L<=m)
ret += getSum(L,R,l,m,rt*2);
if(R>m)
ret += getSum(L,R,m+1,r,rt*2+1);
return ret;
}
int getMax(int L, int R, int l, int r, int rt){ //区间最大值
if(L<=l && r<=R)
return maxn[rt];
int ret = 0;
int m = (l+r)/2;
if(L<=m)
ret = max(ret,getMax(L,R,l,m,rt*2));
if(R>m)
ret = max(ret,getMax(L,R,m+1,r,rt*2+1));
return ret;
}