UOJ #218. 【UNR #1】火车管理

Description

Solution

实际上添加问题就是一个线段树区间覆盖问题,打标记就好
对于弹栈操作比较难搞,实际上也就是一个历史查询,我们不需要保存栈中的每一个元素,我们通过查找历史状态就可以了
这样用主席树维护复杂度是 \(O(n*logn)\) 的
具体是这样的:
假设我们要弹出位置 \(x\) 的栈顶元素,那么在线段树中维护一个值 \(t\),表示最近的一次修改是 \(t\) 时刻
那么上上次修改就可以通过查询 \(t-1\) 时刻的 \(t\) 找出,相当于保存了一个前驱
用主席树维护这个时间就好了

注意内存有些卡,有一些技巧:
1.首先对于查询的线段树是全局的,不需要动态开点
2.对于线段树中的一个节点 \(x\) ,如果它的左右儿子都没有儿子,那么下一次做区间覆盖时,就不需要对 \(x\) 新建两个节点

#include<bits/stdc++.h>
#define lo (o<<1)
#define ro (o<<1|1)
using namespace std;
const int N=5e5+10;
int n,m,ty,rt[N],a[N],tt=0;
struct data{
    int ls,rs,lag;
    data(){lag=-1;}
}tr[N*130];
int T[N*4],la[N*4],in[N*130];
inline void pushdown(int o){
    if(tr[o].lag==-1)return ;
    int t=tr[o].lag;tr[o].lag=-1;
    if(!in[o] || in[tr[o].ls] || in[tr[o].rs]){
        tr[++tt]=tr[tr[o].ls];tr[o].ls=tt;
        tr[++tt]=tr[tr[o].rs];tr[o].rs=tt;in[o]=1;
    }
    int ls=tr[o].ls,rs=tr[o].rs;
    tr[ls].lag=t;tr[rs].lag=t;
}
inline void Push(int o,int l,int r){
    if(la[o]==-1)return ;
    int k=la[o],mid=(l+r)>>1;la[o]=-1;
    T[lo]=k*(mid-l+1);la[lo]=k;
    T[ro]=k*(r-mid);la[ro]=k;
}
inline void upd(int o){T[o]=T[lo]+T[ro];}
inline int qry(int o,int l,int r,int sa,int se){
    if(sa<=l && r<=se)return T[o];
    int mid=(l+r)>>1,ret=0;
    Push(o,l,r);
    if(se<=mid)ret=qry(lo,l,mid,sa,se);
    else if(sa>mid)ret=qry(ro,mid+1,r,sa,se);
    else ret=qry(lo,l,mid,sa,mid)+qry(ro,mid+1,r,mid+1,se);
    upd(o);
    return ret;
}
inline int qt(int x,int l,int r,int sa){
    if(l==r)return tr[x].lag;
    int mid=(l+r)>>1,ret=0;
    pushdown(x);in[x]=1;
    if(sa<=mid)ret=qt(tr[x].ls,l,mid,sa);
    else ret=qt(tr[x].rs,mid+1,r,sa);
    return ret;
}
inline void add(int o,int l,int r,int sa,int se,int t){
    if(sa<=l && r<=se){T[o]=(r-l+1)*t;la[o]=t;return ;}
    Push(o,l,r);
    int mid=(l+r)>>1;
    if(se<=mid)add(lo,l,mid,sa,se,t);
    else if(sa>mid)add(ro,mid+1,r,sa,se,t);
    else add(lo,l,mid,sa,mid,t),add(ro,mid+1,r,mid+1,se,t);
    upd(o);
}
inline void addtag(int &x,int l,int r,int sa,int se,int t){
    tr[++tt]=tr[x];x=tt;
    if(sa<=l && r<=se){tr[x].lag=t;return ;}
    pushdown(x);
    int mid=(l+r)>>1;in[x]=1;
    if(se<=mid)addtag(tr[x].ls,l,mid,sa,se,t);
    else if(sa>mid)addtag(tr[x].rs,mid+1,r,sa,se,t);
    else addtag(tr[x].ls,l,mid,sa,mid,t),addtag(tr[x].rs,mid+1,r,mid+1,se,t);
}
int main(){
    freopen("railway.in","r",stdin);
    freopen("railway.out","w",stdout);
    cin>>n>>m>>ty;
    int op,l,r,ans=0,x,y;
    memset(la,-1,sizeof(la));
    for(int i=1;i<=m;i++){
        rt[i]=rt[i-1];
        scanf("%d%d",&op,&l);
        l=(l+ans*ty)%n+1;
        if(op==1){
            scanf("%d",&r);
            r=(r+ans*ty)%n+1;
            if(l>r)swap(l,r);
            printf("%d\n",ans=qry(1,1,n,l,r));
        }
        else if(op==2){
            x=qt(rt[i],1,n,l);
            if(x){
                y=qt(rt[x-1],1,n,l);
                addtag(rt[i],1,n,l,l,y);add(1,1,n,l,l,a[y]);
            }
        }
        else if(op==3){
            scanf("%d%d",&r,&a[i]);
            r=(r+ans*ty)%n+1;
            if(l>r)swap(l,r);
            add(1,1,n,l,r,a[i]);
            addtag(rt[i],1,n,l,r,i);
        }
    }
    return 0;
}

转载于:https://www.cnblogs.com/Yuzao/p/8698123.html