线段树

线段树

1、线段树是一棵二叉搜索树,它储存的是一个区间的信息。

2、每个节点以结构体的方式存储,结构体包含以下几个信息:

  1.      区间左端点、右端点;(这两者必有)
  2.      这个区间要维护的信息(事实际情况而定,数目不等)。

3、线段树的基本思想:        二分     

4、线段树一般结构如图所示:

 

 

             每个节点的左孩子区间范围为[l,mid],右孩子为[mid+1,r]

 

线段树的基础操作主要有5个:

             建树、单点查询、单点修改、区间查询、区间修改

(以下均为求和操作)

 

 

 

1 struct node{
2     int l,r,w;//l左区间,r右区间,w区间和 
3 }xdtree[4*n];//n个数 

 

一、建树

a、对于二分到的每一个结点,给它的左右端点确定范围。

b、如果是叶子节点,存储要维护的信息。

c、状态合并。

!!::      4倍空间

                    不要漏了return语句

 

 1 void build(int l,int r,int k){//k是根节点的下标
 2     
 3     xdtree[k].l=l;
 4     xdtree[k].r=r;
 5     
 6     if(l==r){
 7         
 8         cout<<xdtree[k].w;//n个数的和 
 9         return; 
10     }
11     
12     int m=(l+r)/2;
13     build(l,m,2*k);
14     build(m+1,r,2*k+1);
15     xdtree[k].w=xdtree[2*k].w+xdtree[2*k+1].w;
16     
17     } 

 

 

 

二、单点查询

 1 void ask(int k){//k一般为1,x是查询点
 2     int ans;
 3     
 4     if(xdtee[k].l==xdtree[k].r)
 5     ans=xdtree[k].w,return;
 6     
 7     int m=(xdtree[k].l + xdtree[k].r)/2;
 8     if(x<=m)  ask(k*2) ;//若目标靠左,递归左儿子 
 9     else ask(k*2+1);//目标靠右。递归右儿子 
10     
11 }

 

三、单点修改

第x个数加上y

 

 

 1 void add(int k){
 2      
 3      if(xdtree[k].l==xdtree[k].r)
 4      xdtree[k].w+=y,return;
 5      
 6      int m=(xdtree[k].l+xdtree[k].r)/2;
 7      if(x<=m) add(k*2);
 8      else add(k*2+1);
 9      
10      xdtree[k].w=xdtree[k*2].w+xdtree[k*2+1].w;//更新包含k的节点 
11 } 
12  

 

四、区间查询

 

查询[ x , y ]的值

 

mid=(l+r)/2

y<=mid ,即 查询区间全在,当前区间的左子区间,往左孩子走

x>mid 即 查询区间全在,当前区间的右子区间,往右孩子走

否则,两个子区间都走

 

 1 void ask_(int k){
 2      
 3     if(xdtree[k].l>=x&&xdtree[k].r<=y)
 4     ans += xdtree[k].w,return;
 5     
 6     int m=(xdtree[k].l+xdtree[k].r)/2;
 7     
 8     if(y>m) ask_(k*2+1);
 9     if(x<=m)  ask_(k*2);
10     
11 } 
12  

 

 

 五、区间修改

 

 修改的时候只修改对查询有用的点。

 引入

   懒标记:存储到这个节点的修改信息,暂时不把修改信息传到子节点。就像家长扣零花钱,你用的时候才给你,不用不给你。

           a.原结构体中增加新的变量,存储这个懒标记。

           b.递归到这个节点时,只更新这个节点的状态,并把当前的更改值累积到标记中。注意是累积,可以这样理解:过年,很多个亲戚都给你压岁钱,但你暂时不用,所以都被你父母扣下了。

           c.什么时候才用到这个懒标记?当需要递归这个节点的子节点时,标记下传给子节点。这里不必管用哪个子节点,两个都传下去。就像你如果还有妹妹,父母给你们零花钱时总不能偏心吧

           d.下传操作:

               3部分:①当前节点的懒标记累积到子节点的懒标记中。

                            ②修改子节点状态。在引例中,就是原状态+子节点区间点的个数*父节点传下来的懒标记

                            这就有疑问了,既然父节点都把标记传下来了,为什么还要乘父节点的懒标记,乘自己的不行吗?

                            因为自己的标记可能是父节点多次传下来的累积,每次都乘自己的懒标记造成重复累积

                             ③父节点懒标记清0。这个懒标记已经传下去了,不清0后面再用这个懒标记时会重复下传。就像你父母给了你5元钱,你不能说因为前几次给了你10元钱, 所以这次给了你15元,那你不就亏大了。 

     懒标记下穿代码:f为懒标记,其余变量与前面含义一致。

 void down(int k)//标记下传 
{
    xdtree[k*2].f+=xdtree[k].f;
    xdtree[k*2+1].f+=xdtree[k].f;
    xdtree[k*2].w+=xdtree[k].f*(xdtree[k*2].r-xdtree[k*2].l+1);
    xdtree[k*2+1].w+=xdtree[k].f*(xdtree[k*2+1].r-xdtree[k*2+1].l+1);
    xdtree[k].f=0;
}

 

 1 void add_(int k){
 2     
 3     if(xdtree[k].l>=a&&xdtree[k].r<=b)
 4     {
 5         xdtree[k].w+=x*(xdtree[k].r-xdtree[k].l+1);
 6         xdtree[k].f+=x;
 7         return;    
 8     }
 9     if(xdtree[k].f)  down(k);
10     int m=(xdtree[k].l+xdtree[k].r)/2;
11     if(y>=m)  add_(k*2);
12     if(x<m)  add_(k*2+1);
13     xdtree[k].w=xdtree[k*2].w+xdtree[k*2+1].w;
14 }

 

 

 

因为引入了懒标记,很多用不着的更改状态存了起来,这就会对区间查询、单点查询造成一定的影响。

所以在使用了懒标记的程序中,单点查询、区间查询也要像区间修改那样,对用得到的懒标记下传。其实就是加上一句if(tree[k].f)  down(k),其余不变。

     引入了懒标记的单点查询代码:

复制代码
 void ask(int k)//单点查询
{
    if(xdtree[k].l==xdtree[k].r)
    {
        ans=xdtree[k].w;
        return ;
    }
    if(xdtree[k].f) down(k);//懒标记下传,唯一需要更改的地方
    int m=(xdtree[k].l+xdtree[k].r)/2;
    if(y>m) ask(k*2+1);
    else ask(k*2);
}
复制代码

    引入了懒标记的区间查询代码:

复制代码
void ask_(int k)//区间查询
{
    if(xdtree[k].l>=x&&xdtree[k].r<=y) 
    {
        ans+=xdtree[k].w;
        return;
    }
    if(xdtree[k].f)  down(k)//懒标记下传,唯一需要更改的地方
    int m=(xdtree[k].l+xdtree[k].r)/2;
    if(y>m) ask_(k*2+1);
    if(x<=m) ask_(k*2);
}
复制代码

\

 

 

 

LAST  空间优化

http://www.cppblog.com/MatoNo1/archive/2015/05/05/195857.html

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

参考:https://www.cnblogs.com/TheRoadToTheGold/p/6254255.html 

 

posted @ 2019-02-22 21:51  MENG_GE  阅读(153)  评论(0编辑  收藏  举报