[SCOI2014]方伯伯的OJ(线段树）

Posted 2021-01-09 zh-comld

方伯伯正在做他的Oj。现在他在处理Oj上的用户排名问题。Oj上注册了n个用户，编号为1～n“，一开始他们按照编号排名。

方伯伯会按照心情对这些用户做以下四种操作，修改用户的排名和编号：

1．操作格式为1 x y，意味着将编号为x的用户编号改为y，而排名不变，执行完该操作后需要输出该用户在队列中的位置，数据保证x必然出现在队列中，同时，1是一个当前不在排名中的编号。

2．操作格式为2 x，意味着将编号为x的用户的排名提升到第一位，执行完该操作后需要输出执行该操作前编号为x用户的排名。

3．操作格式为3 x，意味着将编号为x的用户的排名降到最后一位，执行完该操作后需要输出执行该操作前编号为x用户的排名。

4．操作格式为4 k，意味着查询当前排名为k的用户编号，执行完该操作后需要输出当前操作用户的编号。

但同时为了防止别人监听自己的工作，方伯伯对他的操作进行了加密，即将四种操作的格式分别改为了：

• 1 x+a y+a
• 2 x+a
• 3 x+a
• 4 k+a
• 其中a为上一次操作得到的输出，一开始a=0。

例如：上一次操作得到的输出是5这一次操作的输入为：1 13 15因为这个输入是经过加密后的，所以你应该处理的操作是1 8 10现在你截获了方伯伯的所有操作，希望你能给出结果。

Solution

调了一晚上。。。菜的不行。

因为数列每次只会变化一个数，所以可以不用splay或fhqtreap，用一颗动态开点线段树就可以完成。

因为我们的移动只关于先插和后插，所以我们线段树的区间为m+n+m对于一个移动，我们先把该点在对应位置删除，再将该数插入前/后对应位置。

与此同时，我们开两个map，一个储存编号对应的位置，一个储存位置对应的编号。

Code

#include<iostream>
#include<cstdio>
#include<map>
#define ls tr[cnt].l
#define rs tr[cnt].r
#define N 200002
using namespace std;
map<int,int>mp,anti_mp;
int tot,id,ans,n,m,root,h,t,x,y;
struct seg{
    int l,r,num;
    bool la;
}tr[N*20];
inline int rd(){
    int x=0;char c=getchar();
    while(!isdigit(c))c=getchar();
    while(isdigit(c)){
        x=(x<<1)+(x<<3)+(c^48);
        c=getchar();
    } 
    return x;
}
inline void pushdown(int cnt,int l1,int l2){
    if(!ls)ls=++tot;if(!rs)rs=++tot;
    tr[ls].num=l1;tr[rs].num=l2;
    tr[cnt].la=0;tr[ls].la=tr[rs].la=1;
}
void add(int &cnt,int l,int r,int x,int tag,int rk){
    if(!cnt)cnt=++tot;
    if(l==r){
        if(tag<0)ans=rk+1,anti_mp.erase(l);
        else anti_mp[l]=id;
        tr[cnt].num-=tag;
        return;
    }
    int mid=(l+r)>>1;
    if(tr[cnt].la)pushdown(cnt,mid-l+1,r-mid);
    if(mid>=x)add(ls,l,mid,x,tag,rk);
    else add(rs,mid+1,r,x,tag,rk+mid-l+1-tr[ls].num); 
    tr[cnt].num=tr[ls].num+tr[rs].num;
}
void find(int &cnt,int l,int r,int rk){
    if(!cnt)cnt=++tot;
    if(l==r){
        if(anti_mp.find(l)!=anti_mp.end())ans=anti_mp[l];else ans=l-m;
        return;
    }
    int mid=(l+r)>>1;
    if(tr[cnt].la)pushdown(cnt,mid-l+1,r-mid);
    int num=mid-l+1-tr[ls].num;    
    if(num>=rk)find(ls,l,mid,rk);
    else find(rs,mid+1,r,rk-num);
}
void gai(int &cnt,int l,int r,int L,int R){
    if(!cnt)cnt=++tot;
    if(l>=L&&r<=R){
        tr[cnt].num=r-l+1;
        tr[cnt].la=1;
        return; 
    }
    int mid=(l+r)>>1;
    if(mid>=L)gai(ls,l,mid,L,R);
    if(mid<R)gai(rs,mid+1,r,L,R);
    tr[cnt].num=tr[ls].num+tr[rs].num;
}
void upd(int &cnt,int l,int r,int x,int y,int rk){
    if(!cnt)cnt=++tot;
    if(l==r){
        anti_mp[l]=y;ans=rk+1;
        return;
    }
    int mid=(l+r)>>1;
    if(tr[cnt].la)pushdown(cnt,mid-l+1,r-mid);
    if(mid>=x)upd(ls,l,mid,x,y,rk);
    else upd(rs,mid+1,r,x,y,rk+mid-l+1-tr[ls].num);
}
int main(){
    n=rd();m=rd();
    gai(root,1,n+2*m,1,m);
    gai(root,1,n+2*m,n+m+1,n+m*2);
    h=m+1;t=n+m;
    for(int i=1;i<=m;++i){
        int tag=rd();x=rd()-ans;
        if(tag==1){
            y=rd()-ans;
            int pos;
            if(mp.find(x)!=mp.end())pos=mp[x];
            else pos=x+m;
            mp[y]=pos;
            upd(root,1,n+2*m,pos,y,0);
        }
        else if(tag==2){
            int pos;
            if(mp.find(x)!=mp.end())pos=mp[x];
            else pos=x+m;
            if(anti_mp.find(pos)!=anti_mp.end())id=anti_mp[pos];
            else id=x;
            add(root,1,n+2*m,pos,-1,0);
            add(root,1,n+2*m,--h,1,0);
            mp[x]=h;
        }
        else if(tag==3){
            int pos;
            if(mp.find(x)!=mp.end())pos=mp[x];
            else pos=x+m;
            if(anti_mp.find(pos)!=anti_mp.end())id=anti_mp[pos];
            else id=x;
            add(root,1,n+2*m,pos,-1,0);
            add(root,1,n+2*m,++t,1,0);
            mp[x]=t;
        }
        else find(root,1,n+2*m,x);
        printf("%d
",ans);
    }
    return 0;
}