线段树其实就是把区间划分的操作。递归算法是从上至下,而非递归算法是从下至上。其实线段树和树状数组的思想都差不多,都是因为如果直接存数组的值,那么求和需要O(n),更新需要O(1);如果存数组的和,那么更新需要O(n),求和需要O(1)。既然如此,那就想办法存一组值,使每次只需更新几组相关的数组元素,求和只需加上几组相关的数组元素。
解释一下递归算法为什么要开大概4倍空间,因为左子树和右子树是父节点的2倍和2倍加一所以即便叶子结点那一层没有什么点,如果他们的父节点位于上一层的最右边,那就还需要多开2倍空间。
最小空间:最小的比n大的2的倍数再乘以2。
递归算法如下:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#define maxn 100007 //元素总个数
using namespace std;
int Sum[maxn<<2];//Sum求和,开四倍空间
int A[maxn],n;//存原数组下标[1,n]
//PushUp函数更新节点信息,这里是求和
void PushUp(int rt){Sum[rt]=Sum[rt<<1]+Sum[rt<<1|1];}
//Build函数建立线段树
void Build(int l,int r,int rt){ //[l,r]表示当前节点区间,rt表示当前节点的实际存储位置
if(l==r) {//若到达叶节点
Sum[rt]=A[l];//存储A数组的值
return;
}
int m=(l+r)>>1;
//左右递归
Build(l,m,rt<<1);
Build(m+1,r,rt<<1|1);
//更新信息
PushUp(rt);
}
void Update(int L,int C,int l,int r,int rt){//[l,r]表示当前区间,rt是当前节点编号//l,r表示当前节点区间,rt表示当前节点编号
if(l==r){//到达叶节点,修改叶节点的值
Sum[rt]+=C;
return;
}
int m=(l+r)>>1;
//根据条件判断往左子树调用还是往右
if(L <= m) Update(L,C,l,m,rt<<1);
else Update(L,C,m+1,r,rt<<1|1);
PushUp(rt);//子节点的信息更新了,所以本节点也要更新信息
}
int Query(int L,int R,int l,int r,int rt){//[L,R]表示操作区间,[l,r]表示当前区间,rt:当前节点编号
if(L <= l && r <= R){
//在区间内直接返回
return Sum[rt];
}
int m=(l+r)>>1;
//左子区间:[l,m] 右子区间:[m+1,r] 求和区间:[L,R]
//累加答案
int ANS=0;
if(L <= m) ANS+=Query(L,R,l,m,rt<<1);//左子区间与[L,R]有重叠,递归
if(R > m) ANS+=Query(L,R,m+1,r,rt<<1|1); //右子区间与[L,R]有重叠,递归
return ANS;
}