线段树代码

最新推荐文章于 2023-11-14 19:59:28 发布

米诺minoz

最新推荐文章于 2023-11-14 19:59:28 发布

阅读量385

点赞数 1

分类专栏： ACM ---- 知识点 ACM ---- 数据结构

本文链接：https://blog.csdn.net/JKdd123456/article/details/98504669

版权

ACM ---- 知识点同时被 2 个专栏收录

57 篇文章 1 订阅

订阅专栏

ACM ---- 数据结构

15 篇文章 0 订阅

订阅专栏

以　hdu1166为例(区间求和、单点修改)

一、定义

const LL MAXX = 5e4+10;
int sum[MAXX << 2];
int a[MAXX];

二、区间求和时,更新结点信息


void PushUp(int id)
{
    sum[id] = sum[id<<1] + sum[id<<1|1];
}

三、建树


void Build(int l,int r,int id)
{
    if(l == r){
        sum[id] = a[l];
        return ;
    }
    int mid = (l+r) >> 1;
    Build(l,mid,id<<1);
    Build(mid+1,r,id<<1|1);
    PushUp(id);
}

四、区间求和


int Query(int l,int r,int ll,int rr,int id)
{
    if(ll <= l && r <= rr){　// 如果查询的区间[ll,rr]包含当前区间[l,r],就可以直接加上当前区间的值
        return sum[id];
    }
    int mid = (l+r)>>1;
    int ans = 0;   // 注意清零
    if(ll <= mid)　// 如果[[ll,rr]区间内有结点在左子树
        ans += Query(l,mid,ll,rr,id<<1);
    if(rr > mid)　 // 如果[[ll,rr]区间内有结点在右子树,是 if 而不是 else 
        ans += Query(mid+1,r,ll,rr,id<<1|1);
    return ans;
}

五、单点修改


void Update(int l,int r,int pos,int w,int id)
{
    if(l == r){　// 如果是叶子结点就修改
        sum[id] += w;
        return ;
    }
    int mid = (l+r)>>1;
    if(pos <= mid)　// 如果结点的下标在左子树
        Update(l,mid,pos,w,id<<1);
    else　　　　　　  // 如果结点的下标在右子树
        Update(mid+1,r,pos,w,id<<1|1);
    PushUp(id);  // 如果修改了当前结点pos的值，需要修改与pos相连结点的值，以及一直往上的结点值
}

六、区间更新，延迟标记

(1) 下推标记函数

/**把当前结点的信息更新给儿子结点
   llen rlen 分别代表以 id　为结点的左子树长度和右子树长度
　　对某一个区间进行改变时
　　如果该结点被标记了,在查询的时候就得把改变传给子节点，因为查询的并不一定是当前结点所在的区间
**/

void PushDown(int id,int llen,int rlen)
{
    if(add[id]){// 如果被该结点被标记，则需要将这个标记改变子节点
        add[id<<1] += add[id];
        add[id<<1|1] += add[id];
        sum[id<<1] += add[id]*llen;//该结点id的左孩子长度是 m-l+1
        sum[id<<1|1] += add[id]*rlen;//该结点id的右孩子长度是 r-m
        add[id] = 0;    //将标记向儿子节点移动后，父节点的延迟标记去掉
    }
    // 可能存在多个标记,所以用 += 
}

(2) 查询函数


LL Query(int l,int r,int id,int ll,int rr)
{
　　　// 如果当前区间[l,r]包含在查询区间[ll,rr]内,则直接加上该结点的值，不需要继续去找孩子了
    if(ll <= l && r <= rr)
        return sum[id];

    int mid = (l+r) >> 1;

    //要用id子节点的值时，也要先把id的延迟标记向下移动
    PushDown(id,mid-l+1,r-mid);

    LL ans = 0;
    if(ll <= mid)
        ans += Query(l,mid,id<<1,ll,rr);
    if(rr > mid)
        ans += Query(mid+1,r,id<<1|1,ll,rr);
    return ans;
}

(3) 更新函数


void Update(int l,int r,int id,int ll,int rr,LL w)
{
　　// 如果当前区间[l,r]包含在查询区间[ll,rr]内,则直接加上该结点的值，不需要继续去找孩子了　　　
    if(ll <= l && r <= rr){
        add[id] += w;　　　//给当前结点添加延迟标记
        sum[id] += w*(r-l+1);　　//当前结点的和需要加上 w*长度
        return ;
    }
    int mid = (r+l) >> 1;

    //要对被延迟标记过的这段区间的儿子节点进行更新时，先要将延迟标记向儿子节点移动
    PushDown(id,mid-l+1,r-mid); // 下推延迟标记

    if(ll <= mid)
        Update(l,mid,id<<1,ll,rr,w);
    if(rr > mid)
        Update(mid+1,r,id<<1|1,ll,rr,w);
    PushUp(id);//向上传递更新和
}

解释延迟标记：

对于[3, 9]这样一次修改操作，我可以只去修改如下图中橙色的结点，但是在这个基础上，我要在[3, 9]分解出的4个区间[3, 3], [4, 5], [6, 8], [9, 9]所对应的结点上做一个延迟标记，表示“之前有一次操作需要将这棵子树中的所有结点的价格都进行修改，但是因为还没有用到这棵子树中的值所以我暂时不去修改”

如果修改 [3,9]区间

(代码中被标记的是 3、4-5 、6-8、9)

但是这些懒惰标记（即延迟标记，以下将不在区分二者）又有什么用呢？

其实就是一个暂时不处理，等到需要用到的时候再进行处理的思想。比如你之前说道的询问了[6, 7]这个区间，那么我再从上往下分解的时候，你会发现[6, 8]这个区间上有一个懒惰标记，那么你就应该进行一个懒惰标记的‘下放操作’——也就是说去修改[6, 8]这个结点的左右儿子的值，并且同时给左右儿子添加上新的懒惰标记，然后将[6, 8]的懒惰标记去掉。

也就是说——本来[6, 8]的左右儿子早在之前的修改操作中就需要一同进行修改的，但是因为还没有用到单独的[6, 7]，[8, 8]之类的区间的原因，所以我可以暂时不处理这些结点的修改，而是用一个懒惰标记记录下来这些结点需要进行修改这件事情。然后在之后要用到这些值的时候再进行这些操作。