自己以前学了线段树,感到很懵逼,大概用了一个月的时间才能熟练地打出模板。现在看来,线段树是万能的,这是一个不可不学的数据结构,线段树的应用范围,很广尽管写起来稍微有点烦,代码稍微长了一点。
线段树,顾名思义,就是一棵由线段组成的树他利用了二分的思想,将一个线段以递归的形式二分下去,形成了一棵二叉树。
下面是一棵线段树的模型(本人喜欢左右都闭):
以求和为例,(朴素易懂)代码如下:
struct{
int left,right,sum;
}segment;
segment tree[MAXN*4];
void build(int x,int l,int r){
tree[x].left=l;
tree[x].right=r;
if (l==r){ //到了叶子结点;
tree[x].sum=a[l];
}
else {
int mid=(l+r)/2;
build(x*2,l,mid); //建立左子树 ;
build(x*2+1,mid+1,r);//建立右子树 ;
tree[x].sum=tree[x*2].sum+tree[x*2+1].sum;
}
}
应该比较显然,下面为单点修改,每个点加上一个数:
void add(int x,int i,int data){
int l=tree[x].left;
int r=tree[x].right;
if (l==r&&l==i){ //找到了要修改的点;
tree[x].sum=data;
return;
}
int mid=(l+r)/2;
if (i<=mid) add(x*2,i,data);//如果在左边,就查下去;
else if (i>mid) add(x*2+1,i,data);//如果在右边,继续;
tree[x].sum=tree[x*2].sum+tree[x*2+1].sum; //返回过来后修改父亲节点;
}
很好理解吧,接下来是区间查询(单点查询非常简单,和单点修改差不多,一般没什么用,也不会用),分为四种情况(红色为要查询的区间,黑色为当前区间,画的丑不要介意):
1.
2.
3.
4.
代码如下:
int query(int x,int l,int r){
int le=tree[x].left;
int ri=tree[x].right;
if (l<=le&&r>=r){ // 第一种情况
return tree[x].sum;
}
int mid=(le+ri)/2;
if (l>mid) return query(x*2+1,l,min(r,ri)); //第二种情况
else if (r<=mid) return query(x*2,max(l,le),r); //第三种情况
else return query(x*2,l,mid)+query(x*2+1,mid+1,r); //第四种情况
}
至于多点修改的问题,我即将会给出链接,敬请期待!