数据结构(4)线段树、延迟标记、扫描线

活动 - AcWing

参考-《算法竞赛进阶指南》

一、延迟标记(懒标记)

在线段树的区间查询命令中,每当遇到被查询区间[l,r]完全覆盖节点时,可以直接把节点上的答案作为备选答案返回。我们已经证明,这样操作的复杂度为O(4logn)。

但是在区间修改命令中,如果某个节点被修改区间完全覆盖,那么以该节点为根的整棵子树都要更新,如果逐一更新,时间复杂度为O(n)

试想,如果我们在一次修改指令中发现节点p代表的区间被修改区间完全覆盖,并更新了所有p子树的节点,但是在之后却没有用到这些答案,那么我们的更新就白费了。

因此,我们在执行修改指令时,如果节点被修改区间完全覆盖,我们可以直接回溯,并且在节点上加一个延迟标记,表示“该节点曾被修改,但其子节点没有被更新”。

如果在后面的操作中,如果需要从p往下递归,我们再检查p是否具有标记。如果有更新p的直接子节点,再在子节点上打上标记,然后清除p的标记。

也就是说,除了在修改指令中直接划分成的O(logn)个结点外,对任意节点的修改都延迟到“在后续操作递归进入它的父节点”时进行。这样一来,查询和修改的时间复杂度都是O(logn)级别。

例题1:一个简单的整数问题2

 本题之前用树状数组和差分结合解决过。这里再介绍线段树懒标记做法。

更新懒标记需要用到pushdown函数。(与pushup函数对应)

推导区间和的懒标记pushdown的转移方程:将懒标记加到子节点的标记上,子节点的sum加上子节点区间长度*懒标记。最后清空父节点懒标记

left.add+=root.add,right.add+=root.add;

left.sum+=(LL)(left.r-left.l+1)*root.add;right.sum+=(LL)(right.r-right.l+1)*root.add;

root.add=0;

完整代码:

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;

const int N =1e5+10;
typedef long long LL;

int n,m;
int w[N];
struct Node{
    int l,r;
    LL sum,add;
}tr[4*N];

void pushup(int u)
{
    tr[u].sum=tr[u<<1].sum+tr[u<<1|1].sum;
}
void pushdown(int u)
{
    auto &root=tr[u],&left=tr[u<<1],&right=tr[u<<1|1];
    if(root.add)
    {
        left.add+=root.add,right.add+=root.add;
        left.sum+=(LL)(left.r-left.l+1)*root.add;
        right.sum+=(LL)(right.r-right.l+1)*root.add;
        root.add=0;
    }

}

void build(int u,int l,int r)
{
    if(l==r) tr[u]={l,r,w[l],0};
    else
    {
        tr[u]={l,r};
        int mid=l+r>>1;
        build(u<<1,l,mid),build(u<<1|1,mid+1,r);
        pushup(u);
    }
}

void modify(int u,int l,int r,int d)
{
    if(tr[u].l>=l&&tr[u].r<=r)
    {
        tr[u].sum+=(LL)(tr[u].r-tr[u].l+1)*d;
        tr[u].add+=d;
    }
    else
    {
        pushdown(u);
        int mid=tr[u].l+tr[u].r>>1;
        if(l<=mid) modify(u<<1,l,r,d);
        if(r>mid) modify(u<<1|1,l,r,d);
        pushup(u);
    }

}
LL query(int u,int l,int r)
{
    if(tr[u].l>=l&&tr[u].r<=r) return tr[u].sum;
    pushdown(u);
    int mid=tr[u].l+tr[u].r>>1;
    LL sum=0;
    if(l<=mid) sum+=query(u<<1,l,r);
    if(r>mid) sum+=query(u<<1|1,l,r);
    return sum;
}

int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>w[i];
    build(1,1,n);
    char op[2];
    int l,r,d;

    while(m--)
    {
        cin>>op>>l>>r;
        if(*op=='C')
        {
            cin>>d;
            modify(1,l,r,d);
        }
        else
        {
            cout<<query(1,l,r)<<endl;
        }
    }
    return 0;
}

作者:yankai
链接:https://www.acwing.com/activity/content/code/content/5209661/
来源:AcWing
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

例题2:1277. 维护序列

 处理操作多了一个将区间内每个数乘以一个值。

懒标记处理更加复杂了,对此,我们需要两个懒标记来表示一个操作。一个mul,一个add。

对每一个操作,我们可以把之归为一类操作,即:把区间乘一个数再加一个数,或者把区间加一个数再乘一个数。显然选择抽象成把区间乘一个数再加一个数更为方便。

因为如果要把区间加一个数d,我们可以直接归类为:把区间乘1再加d。

如果把区间乘一个数d,可以归类为:把区间乘d再加0。不需要多余的转化。

因此我们的懒标记pushdown方法方程为:

void eval(Node &t, int add, int mul)
{
    t.sum = ((LL)t.sum * mul + (LL)(t.r - t.l + 1) * add) % p;
    t.mul = (LL)t.mul * mul % p;
    t.add = ((LL)t.add * mul + add) % p;
}
void pushdown(int u)
{
    eval(tr[u<<1],tr[u].add,tr[u].mul);
    eval(tr[u<<1|1],tr[u].add,tr[u].mul);
    tr[u].add=0,tr[u].mul=1;
}

大体框架不变,完整代码为:

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;

const int N =1e6+10;
typedef long long LL;

int n,p,m;
int w[N];
struct Node{
    int l,r;
    LL sum,add,mul;
}tr[N*4];

void pushup(int u)
{
    tr[u].sum=(tr[u<<1].sum+tr[u<<1|1].sum)%p;
}

void eval(Node &t, int add, int mul)
{
    t.sum = ((LL)t.sum * mul + (LL)(t.r - t.l + 1) * add) % p;
    t.mul = (LL)t.mul * mul % p;
    t.add = ((LL)t.add * mul + add) % p;
}
void pushdown(int u)
{
    eval(tr[u<<1],tr[u].add,tr[u].mul);
    eval(tr[u<<1|1],tr[u].add,tr[u].mul);
    tr[u].add=0,tr[u].mul=1;
}
void build(int u,int l,int r)
{
    if(l==r) tr[u]={l,r,w[r],0,1};
    else
    {
        tr[u]={l,r,0,0,1};
        int mid=l+r>>1;
        build(u<<1,l,mid),build(u<<1|1,mid+1,r);
        pushup(u);
    }
}
void modify(int u,int l,int r,int add,int mul)
{
    if(tr[u].l>=l&&tr[u].r<=r)  eval(tr[u],add,mul);
    else
    {
        pushdown(u);
        int mid=tr[u].l+tr[u].r>>1;
        if(l<=mid) modify(u<<1,l,r,add,mul);
        if(r>mid) modify(u<<1|1,l,r,add,mul);
        pushup(u);
    }
}

int query(int u,int l,int r)
{
    if(tr[u].l>=l&&tr[u].r<=r) return tr[u].sum;
    else
    {
        pushdown(u);
        int mid=tr[u].l+tr[u].r>>1;
        int sum=0;
        if(l<=mid) sum=query(u<<1,l,r);
        if(r>mid) sum=(sum+query(u<<1|1,l,r))%p;
        return sum;
    }
}

int main()
{
    cin>>n>>p;
    for(int i=1;i<=n;i++) cin>>w[i];
    build(1,1,n);
    cin>>m;
    while(m--)
    {
        int t,l,r,d;
        cin>>t>>l>>r;
        if(t==1)
        {
            cin>>d;
            modify(1,l,r,0,d);
        }
        else if(t==2)
        {
            cin>>d;
            modify(1,l,r,d,1);
        }
        else cout<<query(1,l,r)<<endl;
    }
    return 0;
}

作者:yankai
链接:https://www.acwing.com/activity/content/code/content/5260636/
来源:AcWing
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

二、线段树+扫描线+二分查找+离散化

 【hdu1542】线段树求矩形面积并 - 拦路雨偏似雪花 - 博客园 (cnblogs.com)

 本人理解不到位,以后在补owo

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
#include<cstdio>
#include<vector>

const int N =1e5+10;

int n;
struct Segment{
    double x,y1,y2;
    int k;
    bool operator <(const Segment &t)const
    {
        return x<t.x;
    }
}seg[2*N];

struct Node{
    int l,r;
    int cnt;
    double len;
}tr[4*2*N];

vector<double> ys;

int find(double y)
{
    // 需要返回vector 中第一个 >= y 的数的下标
    return lower_bound(ys.begin(), ys.end(), y) - ys.begin();
}

void pushup(int u)
{
    if(tr[u].cnt)   tr[u].len=ys[tr[u].r+1]-ys[tr[u].l];
    else if(tr[u].l!=tr[u].r)
    {
        tr[u].len = tr[u << 1].len + tr[u << 1 | 1].len;
    }
    else tr[u].len = 0;//表示为叶子节点且该线段没被覆盖,为无用线段,长度变为0
}

void modify(int u,int l,int r,int d)
{
    if(tr[u].l >= l && tr[u].r <= r)//该区间被完全覆盖
    {
        tr[u].cnt += d;//该区间出现次数 + d
        pushup(u);//更新该节点的len
    }
    else
    {
        int mid = tr[u].r + tr[u].l >> 1;
        if (l <= mid)modify(u << 1,l,r,d);//左边存在点
        if (r > mid)modify(u << 1 | 1,l,r,d);//右边存在点
        pushup(u);//进行更新
    }
}
void build(int u,int l,int r)
{
    tr[u] = {l,r,0,0};

    if (l != r)
    {
        int mid = l + r >> 1;
        build(u << 1,l,mid),build(u << 1 | 1,mid + 1,r);
        //后面都为0,不需更新len
    }
}

int main()
{
    int T=1;
    while(scanf("%d",&n),n)
    {
        ys.clear();
        for(int i=0,j=0;i<n;i++)
        {
            double x1,y1,x2,y2;
            scanf("%lf%lf%lf%lf", &x1, &y1, &x2, &y2);
            seg[j ++ ] = {x1, y1, y2, 1};
            seg[j ++ ] = {x2, y1, y2, -1};
            ys.push_back(y1), ys.push_back(y2);   
        }
        sort(ys.begin(),ys.end());
        ys.erase(unique(ys.begin(),ys.end()),ys.end());
        build(1,0,ys.size()-2);
        sort(seg,seg+2*n);
        double res=0;
        for(int i=0;i<n*2;i++)
        {
            if(i>0) res+=tr[1].len*(seg[i].x-seg[i-1].x);
            modify(1, find(seg[i].y1), find(seg[i].y2) - 1, seg[i].k);
        }

        printf("Test case #%d\n", T ++ );
        printf("Total explored area: %.2lf\n\n", res);

    }
    return 0;

}

作者:yankai
链接:https://www.acwing.com/activity/content/code/content/5257208/
来源:AcWing
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

 

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值