萌新刚学线段树,照着深进打了份模板,50pts求调,玄关

P1253 扶苏的问题

chen_z @ 2023-07-20 09:50:19

#include<bits/stdc++.h>
#define int long long
#define inf 1000000000000000000ll
using namespace std;
const int maxn=1000000;
int a[maxn],w[maxn*4],lzy_set[maxn*4],lzy_add[maxn*4];
void pushup(const int u){
    w[u]=max(w[u*2],w[u*2+1]);
}
void build(int u,int L,int R){
    if(L==R){
        w[u]=a[L];
        return;
    }
    int M=(L+R)/2;
    build(u*2,L,M);
    build(u*2+1,M+1,R);
    pushup(u);
    lzy_set[u]=inf;
}
bool InRange(int L,int R,int l,int r){
    return (l<=L)&&(R<=r);
}
bool OutofRange(int L,int R,int l,int r){
    return (L>r)||(R<l);
}
void maketag(int u,int x,int type){
    if(type==1){
        lzy_add[u]=0;
        lzy_set[u]=x;
        w[u]=x;
    }
    else{
        if(lzy_set[u]==inf)lzy_add[u]+=x;
        else lzy_set[u]+=x;
        w[u]+=x;
    }
} 
void pushdown(int u){
    if(lzy_set[u]==inf){
        maketag(u*2,lzy_add[u],2);
        maketag(u*2+1,lzy_add[u],2);
        lzy_add[u]=0;
    }
    else{
        maketag(u*2,lzy_set[u],1);
        maketag(u*2+1,lzy_set[u],1);
        lzy_set[u]=inf;
    }
}
int query(int u,int L,int R,int l,int r){
    if(InRange(L,R,l,r)){
        return w[u];
    }
    else if(!OutofRange(L,R,l,r)){
        int M=(L+R)/2;
        pushdown(u);
        return max(query(u*2,L,M,l,r),query(u*2+1,M+1,R,l,r));
    }
    else return 0;
}
void update(int u,int L,int R,int l,int r,int x,int type){
    if(InRange(L,R,l,r)){
        maketag(u,x,type);
    }
    else if(!OutofRange(L,R,l,r)){
        int M=(L+R)/2;
        pushdown(u);
        update(u*2,L,M,l,r,x,type);
        update(u*2+1,M+1,R,l,r,x,type);
        pushup(u);
    }
}
signed main(){
    int n,q;
    cin>>n>>q;
    for(int i=1;i<=n;i++)cin>>a[i];
    build(1,1,n);
    for(int t=1;t<=q;t++){
        int op,x,y,k;
        cin>>op;
        if(op==1||op==2){
            cin>>x>>y>>k;
            update(1,1,n,x,y,k,op);
        }
        else{
            cin>>x>>y;
            cout<<query(1,1,n,x,y)<<'\n';
        }
    }
    return 0;
}

by chen_z @ 2023-07-20 09:51:33

4 WA 是负数问题,还有最后一个点 T 了,太菜了,不会调


by tangyigeng @ 2023-07-20 17:13:31

首先是cin的问题,cin非常慢,尽量用scanf或者快读。然后是query里应该返回-inf而不是0这问题不知道为什么一开始我也犯了。


by chen_z @ 2023-07-21 19:17:46

@tangyigeng thx


by chen_z @ 2023-07-21 19:22:28

@tangyigeng A了!再次感谢!orz


|