[联合集训6-18] 栈 吉司机线段树

这个做法很巧妙。
考虑一个单调栈的插入过程。假如把插入的数倒过来,依次和栈底元素chkmax,如果chkmax成功就把它放入栈底,最终等效于顺序插入的结果。
于是我们把所有操作倒序处理,假如一个单调栈被询问了 x x 次,我们就在线段树上给其分配 x x 个下标,也就是说最后线段树一共是询问个数个下标,每个下标对应的是某个单调栈的某次询问。
那么用吉司机线段树维护区间chkmax,遇到一个询问就把对应的下标清零即可。具体地,要维护一个加法标记add,当chkmax成功且只改变了最小值(就是没有发生合并,不用向下递归的情况)后,要把原先的最小值加到add中去,那么做完所有操作后,线段树上每个叶子节点的add加上最小值就是对应询问答案。
代码:


#include
#include
#include
#include
#define N 200010
#define ll long long
#define mid (l+r>>1)
#define lc(v) (v<<1)
#define rc(v) (v<<1|1)
#define ll long long
using namespace std;
const int inf=0x3f3f3f3f;
int n,m,cnt[N],pnt[N];
ll ans[N];
int read()
{
    int x=0,f=1;char ch=getchar();
    for(;ch<'0'||ch>'9';ch=getchar()) if(ch=='-') f=-1;
    for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
    return x*f;
}
struct node
{
    int opt,l,r,x;
}q[N];

struct tree
{
    int mi[N<<2],se[N<<2];
    ll add[N<<2];
    void update(int v)
    {
        int x=lc(v),y=rc(v);
        if(mi[x]==mi[y]){mi[v]=mi[x];se[v]=min(se[x],se[y]);return ;}
        if(mi[x]>mi[y]) swap(x,y);
        mi[v]=mi[x];se[v]=min(se[x],mi[y]);
    }
    void pushdown(int v)
    {
        if(mi[lc(v)]lc(v)]=mi[v],add[lc(v)]+=add[v];
        if(mi[rc(v)]0;
    }
    void cal(int v,int d)
    {
        if(mi[v]>=d) return ;
        if(dreturn ;}
        pushdown(v),cal(lc(v),d),cal(rc(v),d);
        update(v);  
    }
    void build(int v,int l,int r)
    {
        if(l==r){mi[v]=add[v]=0;se[v]=inf;return ;}
        build(lc(v),l,mid);
        build(rc(v),mid+1,r);
        update(v);
    }
    void clr(int v,int l,int r,int x)
    {
        if(l==r){mi[v]=add[v]=0;se[v]=inf;return ;}
        pushdown(v);
        if(x<=mid) clr(lc(v),l,mid,x);
        else clr(rc(v),mid+1,r,x);
        update(v);
    }
    void mdf(int v,int l,int r,int lx,int rx,int d)
    {
        if(l==lx&&r==rx){cal(v,d);return ;}
        pushdown(v);
        if(rx<=mid) mdf(lc(v),l,mid,lx,rx,d);
        else if(lx>mid) mdf(rc(v),mid+1,r,lx,rx,d);
        else mdf(lc(v),l,mid,lx,mid,d),mdf(rc(v),mid+1,r,mid+1,rx,d);
        update(v);
    }
    void travel(int v,int l,int r)
    {
        if(l==r) {ans[l]=add[v]+mi[v];return ;}
        pushdown(v);
        travel(lc(v),l,mid);
        travel(rc(v),mid+1,r);
        update(v);
    }
}T;
int main()
{
    freopen("stack.in","r",stdin);
    freopen("stack.out","w",stdout);
    n=read();m=read();
    for(int i=1;i<=m;i++)
    {
        q[i].opt=read();
        if(q[i].opt==1) q[i].l=read(),q[i].r=read(),q[i].x=read();
        else q[i].x=read(),cnt[q[i].x]++;
    }
    for(int i=1;i<=n;i++)
        cnt[i]+=cnt[i-1],pnt[i]=cnt[i-1]+1;
    int all=cnt[n]; 
    T.build(1,1,all);   

    for(int i=m;i;i--)
        if(q[i].opt==1)
        {
            if(cnt[q[i].l-1]q[i].r])T.mdf(1,1,all,cnt[q[i].l-1]+1,cnt[q[i].r],q[i].x);
        }
        else
            T.clr(1,1,all,pnt[q[i].x]),pnt[q[i].x]++;   
    T.travel(1,1,all);      
    for(int i=1;i<=m;i++)
        if(q[i].opt==2)
            printf("%lld\n",ans[--pnt[q[i].x]]);

    return 0;
}

你可能感兴趣的:(线段树)