洛谷P1253 [yLOI2018] 扶苏的问题 （线段树）

一道用来练习打标记的好题。

对于区间加和区间赋值两个操作分别用两个标记，分析如何打标记并下传标记（还是比较好分析的）。

坑点：查询操作时，我一开始把ans设为-0x3f3f3f3f（调试了好久才发现），这显然是不对的（看题目的数据设置），将其赋值为-1e18就行了。

代码中注释的内容是另一种打标记的方法，未打标记的是自己想出来的（自认为更好懂一些）。

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10; 
typedef long long ll;
struct node{
    ll mmax,l,r;
    ll tag1,tag2;//修改，加 
}t[N<<2];
ll n,q,a[N];

void up(ll k){
    t[k].mmax=max(t[k<<1].mmax,t[k<<1|1].mmax);
}

void rev(ll k,ll x,ll y){
    if(x==1e9+1){
        t[k].tag2+=y;
        t[k].mmax+=y;
    }
    else{
        t[k].tag1=x;
        t[k].tag2=y;
        t[k].mmax=x+y;
    }
}

void pd(ll k){
    rev(k<<1,t[k].tag1,t[k].tag2);
    rev(k<<1|1,t[k].tag1,t[k].tag2);
    t[k].tag1=1e9+1;
    t[k].tag2=0;
}

/*void downtag1(ll k){
    if(t[k].tag1!=1e9+1){
        t[k<<1].mmax=t[k<<1].tag1=t[k].tag1;
        t[k<<1].tag2=0;
        t[k<<1|1].mmax=t[k<<1|1].tag1=t[k].tag1;
        t[k<<1|1].tag2=0;
        t[k].tag1=1e9+1;
    }
}

void downtag2(ll k){
    t[k<<1].tag2+=t[k].tag2;
    t[k<<1].mmax+=t[k].tag2;
    t[k<<1|1].tag2+=t[k].tag2;
    t[k<<1|1].mmax+=t[k].tag2;
    t[k].tag2=0;
}*/

void build(ll k,ll l,ll r){
    t[k].l=l,t[k].r=r,t[k].tag1=1e9+1;
    if(l==r){
        t[k].mmax=a[l];    
        return ;
    }
    ll mid=(l+r)>>1;
    build(k<<1,l,mid);build(k<<1|1,mid+1,r);
    up(k);
}

void change1(ll k,ll l,ll r,ll x){
    if(t[k].l>=l && t[k].r<=r){
        t[k].mmax=x;
        t[k].tag1=x,t[k].tag2=0;
        return ;
    }
    pd(k);
    //downtag1(k),downtag2(k);
    ll mid=(t[k].l+t[k].r)>>1;
    if(l<=mid) change1(k<<1,l,r,x);
    if(r>mid) change1(k<<1|1,l,r,x);
    up(k);
}

void change2(ll k,ll l,ll r,ll x){
    if(t[k].l>=l && t[k].r<=r){
        t[k].mmax+=x;
        t[k].tag2+=x;
        return ;
    }
    pd(k);
    //downtag1(k),downtag2(k);
    ll mid=(t[k].l+t[k].r)>>1;
    if(l<=mid) change2(k<<1,l,r,x);
    if(r>mid) change2(k<<1|1,l,r,x);
    up(k);
} 

ll query(ll k,ll l,ll r){
    if(t[k].l>=l && t[k].r<=r) return t[k].mmax;
    pd(k);
    //downtag1(k),downtag2(k);
    ll mid=(t[k].l+t[k].r)>>1;
    ll ans=-1e18;
    if(l<=mid) ans=max(ans,query(k<<1,l,r));
    if(r>mid) ans=max(ans,query(k<<1|1,l,r));
    return ans;
}

int main(){
    scanf("%lld%lld",&n,&q);
    for(int i=1;i<=n;i++)
        scanf("%lld",&a[i]);
    build(1,1,n);
    while(q--){
        ll op,l,r,x;
        scanf("%lld%lld%lld",&op,&l,&r);
        if(op==1){
            scanf("%lld",&x);
            change1(1,l,r,x); 
        }
        else if(op==2){
            scanf("%lld",&x);
            change2(1,l,r,x);
        }
        else cout<<query(1,l,r)<<endl;
    }
}

 

---

如果您觉得阅读本文对您有帮助，请点一下“推荐”按钮，您的“推荐”将是我最大的写作动力！欢迎各位转载，但是未经作者本人同意，转载文章之后必须在文章页面明显位置给出作者和原文连接，否则保留追究法律责任的权利。