吉司机线段树学习笔记

先放例题，复杂度不会整坑了暂时

P6242 【模板】线段树 3

题解都放代码里了

调了半个月

/*
吉司机线段树 
*/
/*
原来的想法是开一个加标记和一个取min标记，发现懒标记下传只能迭代不能合并
考虑到每次做标记的点满足semx<w<mxA，那么可以开两种加的懒标记
一种是最大值的懒标记，一种是非最大值的懒标记，把两种修改值区分 
*/
/*
为啥要开四个懒标记呢
因为对于懒标记传给孙子而言，懒标记的正负会叠加
这时让mxB=max(mxB,mxA) 就是叠加之后去max
忽略了中间只保留正的懒标记的最大值 
*/
/*
新加的两个懒标记是对于原来的懒标记的历史取max
而不是对于历史最大值 非最大值的懒标记 
*/
#include<bits/stdc++.h> 
using namespace std;
#define LL long long
const int MAXN=5e5+100;
const LL INF=1e18;
inline int read(){
    int s=0,w=0;char ch=getchar();
    while(ch<'0'||ch>'9')w|=(ch=='-'),ch=getchar();
    while(ch>='0'&&ch<='9')s=s*10+ch-'0',ch=getchar();
    return w?-s:s;
}
#define kd (read())
int n,m;
struct Segment_tree{
    #define ls (u<<1)
    #define rs (u<<1|1)
    LL mxB[MAXN<<2],mxA[MAXN<<2];
    LL sm[MAXN<<2],semx[MAXN<<2],cnt[MAXN<<2];
    LL lb1[MAXN<<2],lb2[MAXN<<2];
    LL hlb1[MAXN<<2],hlb2[MAXN<<2];
    inline void up(int u){ 
        mxA[u]=max(mxA[ls],mxA[rs]);
        sm[u]=sm[ls]+sm[rs];
        mxB[u]=max(mxB[u],mxA[u]);
        
        if(mxA[ls]==mxA[rs])cnt[u]=cnt[ls]+cnt[rs],semx[u]=max(semx[ls],semx[rs]);
        else if(mxA[ls]>mxA[rs])cnt[u]=cnt[ls],semx[u]=max(mxA[rs],semx[ls]);
        else cnt[u]=cnt[rs],semx[u]=max(mxA[ls],semx[rs]);
        
        return ;
    }
    inline void down(int u,int l,int r){
        if(l==r)return ;
        int mid=l+r>>1;
        int lmx=max(mxA[ls],mxA[rs]);
        
        if(mxA[ls]==lmx){
            //LL t=max(0LL,lb1[ls]+hlb1[u]-hlb1[ls]);
            hlb1[ls]=max(hlb1[ls],lb1[ls]+hlb1[u]);
            mxB[ls]=max(mxB[ls],mxA[ls]+hlb1[u]);
            /*
            
                    这里 因为我更新ch时，ch的懒标记是必须加进ch里的
                    所以我不能用t，这样的话会造成本来必须加进去的部分出错 
            
            */ 
//            mxB[ls]=max(mxB[ls],mxA[ls]+t); 原来错的地方 
            mxA[ls]+=lb1[u];
            sm[ls]+=cnt[ls]*lb1[u];
            lb1[ls]+=lb1[u];
        }
        else{
//            LL t=max(0LL,lb1[ls]+hlb2[u]-hlb1[ls]);
            hlb1[ls]=max(hlb1[ls],lb1[ls]+hlb2[u]);
            mxB[ls]=max(mxB[ls],mxA[ls]+hlb2[u]);
            mxA[ls]+=lb2[u];
            sm[ls]+=cnt[ls]*lb2[u];
            lb1[ls]+=lb2[u];
        }
        hlb2[ls]=max(hlb2[ls],lb2[ls]+hlb2[u]);
        semx[ls]+=lb2[u];
        sm[ls]+=(mid-l+1-cnt[ls])*lb2[u];
        lb2[ls]+=lb2[u];
        
        if(mxA[rs]==lmx){
//            LL t=max(0LL,lb1[rs]+hlb1[u]-hlb1[rs]);
            hlb1[rs]=max(hlb1[rs],lb1[rs]+hlb1[u]);
            mxB[rs]=max(mxB[rs],mxA[rs]+hlb1[u]);
            mxA[rs]+=lb1[u];
            sm[rs]+=cnt[rs]*lb1[u];
            lb1[rs]+=lb1[u];
        }
        else{
//            LL t=max(0LL,lb1[rs]+hlb2[u]-hlb1[rs]);
            hlb1[rs]=max(hlb1[rs],lb1[rs]+hlb2[u]);
            mxB[rs]=max(mxB[rs],mxA[rs]+hlb2[u]);
            mxA[rs]+=lb2[u];
            sm[rs]+=cnt[rs]*lb2[u];
            lb1[rs]+=lb2[u];
        }
        hlb2[rs]=max(hlb2[rs],lb2[rs]+hlb2[u]);
        semx[rs]+=lb2[u];
        sm[rs]+=(r-mid-cnt[rs])*lb2[u];
        lb2[rs]+=lb2[u];
        
        hlb1[u]=hlb2[u]=lb1[u]=lb2[u]=0;
        return up(u),void();
    }
    inline void build(int u,int l,int r){
        hlb1[u]=hlb2[u]=lb1[u]=lb2[u]=0;
        semx[u]=mxA[u]=mxB[u]=-INF;
        cnt[u]=0;sm[u]=0;
        if(l==r)return mxB[u]=mxA[u]=sm[u]=kd,cnt[u]=1,void();
        int mid=l+r>>1;
        build(ls,l,mid),build(rs,mid+1,r);
        return up(u),void();
    }
    inline void modify_j(int u,int l,int r,int x,int y,const LL &w){
        down(u,l,r);
        if(l>=x&&r<=y){
            lb1[u]=w;
            lb2[u]=w;
            hlb1[u]=w;
            hlb2[u]=w;
            mxA[u]+=w;
            semx[u]+=w;
            sm[u]+=w*(r-l+1);
            mxB[u]=max(mxA[u],mxB[u]);
            return ;
        }
        int mid=l+r>>1;
        if(x<=mid)modify_j(ls,l,mid,x,y,w);
        if(y>=mid+1)modify_j(rs,mid+1,r,x,y,w);
        return up(u),void();
    }
    inline void modify_mn(int u,int l,int r,int x,int y,const LL &w){
        down(u,l,r);
        if(mxA[u]<=w)return ;
        if(l>=x&&r<=y&&w>semx[u]){
            sm[u]=sm[u]-cnt[u]*(mxA[u]-w);
            lb1[u]=w-mxA[u];
            mxA[u]=w;
            mxB[u]=max(mxA[u],mxB[u]);
            return ;
        }
        int mid=l+r>>1;
        if(x<=mid)modify_mn(ls,l,mid,x,y,w);
        if(y>=mid+1)modify_mn(rs,mid+1,r,x,y,w);
        return up(u),void();
    }
    inline LL query_mxA(int u,int l,int r,int x,int y){
        down(u,l,r);
        if(l>=x&&r<=y)return mxA[u];
        int mid=l+r>>1;LL res=-INF;
        if(x<=mid)res=max(res,query_mxA(ls,l,mid,x,y));
        if(y>=mid+1)res=max(res,query_mxA(rs,mid+1,r,x,y));
        up(u);
        return res;
    }
    inline LL query_mxB(int u,int l,int r,int x,int y){
        down(u,l,r);
        if(l>=x&&r<=y)return mxB[u];
        int mid=l+r>>1;LL res=-INF;
        if(x<=mid)res=max(res,query_mxB(ls,l,mid,x,y));
        if(y>=mid+1)res=max(res,query_mxB(rs,mid+1,r,x,y));
        up(u);
        return res;
    }
    inline LL query_sm(int u,int l,int r,int x,int y){
        down(u,l,r);
        if(l>=x&&r<=y)return sm[u];
        int mid=l+r>>1;LL res=0;
        if(x<=mid)res+=query_sm(ls,l,mid,x,y);
        if(y>=mid+1)res+=query_sm(rs,mid+1,r,x,y);
        up(u);
        return res;
    }
}T;
int main(){
//    freopen("data.in","r",stdin);
//    freopen("std.out","w",stdout);
    n=kd,m=kd;
    T.build(1,1,n);
    for(int i=1,op,l,r,k;i<=m;++i){
        op=kd,l=kd,r=kd;
        if(op==1)k=kd,T.modify_j(1,1,n,l,r,k);
        else if(op==2)k=kd,T.modify_mn(1,1,n,l,r,k);
        else if(op==3)printf("%lld
",T.query_sm(1,1,n,l,r));
        else if(op==4)printf("%lld
",T.query_mxA(1,1,n,l,r));
        else printf("%lld
",T.query_mxB(1,1,n,l,r));
    }
    return 0;
}
/*
6 6
124 53 171 50 203 152 
3 2 2
2 6 6 167
5 5 6
3 5 6
4 2 6
5 6 6

*/