这是线段树的样子,线段树是一种二叉搜索树,它将一个区间划分成一些单元,每个单元对应线段树的一个叶节点,它是平衡二叉树,最后的子节点数目为N,即区间的长度。它的查询,更新的时间复杂度是O(logn)
线段树的基本操作:
1.建树
2.查询
3.点更新
4.区间更新
5.区间合并
建树过程
//这里的m都是从1开始
struct node{
int l,r,sum;//左右结点
}
void pushup(int m){
tr[m].sum =tr[m<<1].sum+tr[m<<1|1].sum;
}
void BuildTree(int m,int l,int r){//m初始值为1
tr[m].l=l;
tr[m].r=r;
if(l==r){//叶节点
scanf("%d",&tr[m].sum);
return;
}
int mid = (l+r)>>1;
build(m<<1,l,mid);//左节点的序数是根节点序数的两倍
build(m<<1|1,mid+1,r);
pushup(m);//求m的sum和
}
单点更新
void update(int m,int index,int val){
if(tr[m].l==index&&tr[m].r==index){
tr[m].sum+=index;
return;
}
int mid = (tr[m].l+tr[m].r)>>1;
if(index<=mid)update(m<<1,index,val);
else update(m<<1|1,index,val);
pushup(m);//因为单点更新完之后sum的值也得变
}
单点查询
int query(int m,int l,int r){
if(tr[m].l==l&&tr[m].r==r){
return tr[m].sum;
}
int mid = (tr[m].l+tr[m].r)>>1;
int temp;
if(r<=mid)temp=query(m<<1,l,r);
else if(l>mid)temp = query(m<<1|1,l,r);
else temp = query(m<<1,l,mid)+query(m<<1|1,mid+1,r);
return temp;
}
区间更新:
区间更新得有个lazy标记,也就是说对更新的区间进行标记,如果查询更深层的叶节点,如果有Lazy标记,那么就把lazy标记的数往下赋值给下个结点。
struct node{
int l,r,sum;
int lazy;//标记此节点每单位增长的数值
}
void pushdown(int m){
if(tr[m].lazy){
tr[m<<1].lazy+=tr[m].lazy;
tr[m<<1|1].lazy+=tr[m].lazy;
tr[m<<1].sum+=tr[m].lazy*(tr[m<<1].r-tr[m<<1].l+1);
tr[m<<1|1].sum+=tr[m].lazy*(tr[m<<1|1].r-tr[m<<1|1].l+1);
tr[m].lazy=0;//一定要把根节点lazy数值归0
}
//因为都是子节点,所以要一视同仁
}
void update(int m, int l, int r, int val){
if(tr[m].l==l&&tr[m].r==r){
tr[m].sum+=val*(r-l+1);
tr[m].lazy +=val;
return;
}//如果是第一次区间更新的话例如上图1-10,那么就直接添加lazy标记即可,
//否则要把上个根节点的lazy标记往下传
pushdown(m);
int mid =(tr[m].l+tr[m].r)>>1;
if(r<=mid)update(m<<1,l,r,val);
else if(l>mid)update(m<<1|1,l,r,val);
else {
update(m<<1,l,mid,val);
update(m<<1|1,mid+1,r,val);
}
pushup(m);
}
区间查询:
int query(int m,int l,int r){
if(tr[m].l==l&&tr[m].r==r){
return tr[m].sum;
}
pushdown(m);//就是上面区间更新pushdown()
int mid = (tr[m].l+tr[m].r)>>1;
int tmep;
if(r<=mid)temp =query(m<<1,l,r);
else if(l>mid)temp = query(m<<1|1,l,r);
else temp= query(m<<1,l,mid)+query(m<<1|1,mid+1,r);
return temp;
}