码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2018-09-27 22:04:04      阅读:245      评论:0      收藏:0      [点我收藏+]

标签:防止   end   工作   插入   ios   ant   std   bool   处理   

方伯伯正在做他的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\n",ans);
    }
    return 0;
}

 

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

标签:防止   end   工作   插入   ios   ant   std   bool   处理   

原文地址:https://www.cnblogs.com/ZH-comld/p/9715245.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!