线段树
将数组不断二分成若干个段,然后将各个段的和储存在这个树状数组里面。
因为这个线段树组满足下图(画的有点丑)这种规律所以方便我们存数(数字都是下标)
建树与整合
首先我们先开一个要读入的普通数组和线段树的数组
#define MAX 10000
int a[MAX];
struct QAQ{
//l表示此点所对应的左孩子,r表示右孩子
int l, r, sum;//sum表示l到r的区间和
}tree[MAX*4];
//开4倍是因为线段树组存了各个区间的sum区间肯定比原数组大
在主函数读入数组之后,我们就开始建树, 所以我们需要一个建树函数
//建树l为左边界,r为建树右边界, i为此时的线段树下标
void pushup(int i);
void build(int l, int r, int i){
tree[i].1=l,tree[i].r=r;
if(l==r){
tree[i].sum=a[l];
}
int mid = (l + r) / 2;
build(l, mid, i * 2);
build(mid + 1, r, i * 2 + 1);
pushup(i);
}
//我们需要一个pushup函数将下面两个区间的和整合到上面那个区间里
void pushup(int i){
tree[i].sum=tree[i*2].sum+tree[i*2+1].sum;
}
更新
在实际操作中我们需要更新这个树,一般的方法是先找到树的叶节点,再从叶节点回溯到根节点从而更新整棵树。于是我们就需要一个更新函数
//更新点的话我们有需要更新的点的下标和值
//
void update(int l, int r, int i, int idx, int val){
if(l==r){//其实这里r=l=idx
a[idx]=val;
tree[l]=val;
return;
}
int mid= (l+r) / 2;
if(idx<=mid){
update(l, mid, i*2, idx, val);
}else{
update(mid+1, r, i*2+1, idx, val);
}
pushup(i);
}
求区间和
然后到了解决区间问题的时候了, 问一个区间的和,首先我们要有所求区间的左边界和右边界
//L为左边界,R为右边界
int query(int l, int r, int i, int L, int R){
int mid = (l+r) / 2;
if(r<L||l>R){
return 0;
}else if(l>=L&&r<=R||l==r){
return tree[i];
}else {
int left=query(l, mid, i*2, L, R);
int right=quert(mid + 1, r, i*2+1, L, R);
return left + right;
}
}
以上便是一般的线段树结构所拥有的操作,根据不同提型进行变换