[SCOI2014]方伯伯的OJ

看到这道题的第一想法就是要用FHQ treap 过了这道题...于是至今尚未成功(华丽的 T 掉了 (╯‵□′)╯︵┻━┻ )。于是附个地址。

然后水一波博客。

题意简介

emmmm...方伯伯脑抽做了个 oj ,然后想要在对 oj 上的 1~n 编号的用户乱来(并且还对他的乱来操作进行了加密)。你需要维护一棵平衡树完成方伯伯的一波操作

你的平衡树需要支持这些操作: 1. 修改编号; 2. 把一个人放到树的最左边; 3. 把一个人放到树的最右边;4.输出一个排名对应的编号。

题目分析

然后这道题非常坑,要用 map 记录每个节点对应区间右端点的对应节点编号 (听起来很绕,其实就是: 设某节点对应区间右端点为 R , 该节点编号为 p , 那么我们用的map 的 first key 就是 R ,second key 是 p),然后用找某个编号所在的节点用 lower_bound 就好了(才知道map是可以 lower bound 的,并且是以第一关键字作比较的)。我记得有看到用另一种利用 map 的方法,那好像是 普通 treap (splay应该也能用啊)的,而且节点维护的区间也有点混乱...就是这个

Splay 做法

  1 //by Judge
  2 #include<iostream>
  3 #include<cstdio>
  4 #include<map>
  5 using namespace std;
  6 const int M=5e5+111;
  7 //#define getchar() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)
  8 char buf[1<<21],*p1=buf,*p2=buf;
  9 inline int read(){
 10     int x=0,f=1; char c=getchar();
 11     for(;!isdigit(c);c=getchar()) if(c=='-') f=-1;
 12     for(;isdigit(c);c=getchar()) x=x*10+c-'0'; return x*f;
 13 }
 14 char sr[1<<21],z[20];int C=-1,Z;
 15 inline void Ot(){fwrite(sr,1,C+1,stdout),C=-1;}
 16 inline void print(int x){
 17     if(C>1<<20)Ot();if(x<0)sr[++C]=45,x=-x;
 18     while(z[++Z]=x%10+48,x/=10);
 19     while(sr[++C]=z[Z],--Z);sr[++C]='
';
 20 }
 21 int n,m,cnt,ans,root;
 22 map<int,int> mp;
 23 struct Node{
 24     int l,r,siz,fa,ch[2];
 25 }t[M];
 26 inline int newnode(int x,int y){ int u=++cnt; t[u].siz=y-x+1,t[u].l=x,t[u].r=y; return u; }
 27 inline void pushup(int x){ t[x].siz=t[t[x].ch[0]].siz+t[t[x].ch[1]].siz+t[x].r-t[x].l+1; }
 28 inline void rotate(int x){
 29     int y=t[x].fa,z=t[y].fa,sn=t[y].ch[1]==x;
 30     t[x].fa=z; if(z) t[z].ch[t[z].ch[1]==y]=x;
 31     t[y].ch[sn]=t[x].ch[!sn],t[t[y].ch[sn]].fa=y;
 32     t[y].fa=x,t[x].ch[!sn]=y,pushup(y);
 33 }
 34 inline void splay(int x,int to){
 35     while(t[x].fa^to){
 36         int y=t[x].fa,z=t[y].fa;
 37         if(z!=to) rotate((t[z].ch[0]==y)^(t[y].ch[0]==x)?x:y);
 38         rotate(x);
 39     } pushup(x); if(!to) root=x;
 40 }
 41 inline int query(int x){ splay(x,0); return t[x].siz-t[t[x].ch[1]].siz; }
 42 inline int get_id(int k){  //查询排名为 k 的人的编号
 43     int now=root;
 44     while(k){ 
 45         int sum=t[t[now].ch[0]].siz+t[now].r-t[now].l+1;
 46         if(t[t[now].ch[0]].siz<k && k<=sum){
 47             k-=t[t[now].ch[0]].siz; break;
 48         } else if(sum<k) k-=sum,now=t[now].ch[1];
 49         else now=t[now].ch[0];
 50     } return t[now].l+k-1;
 51 }
 52 inline void erase(int x){  //删除节点信息
 53     int pre=t[x].ch[0],nxt=t[x].ch[1];
 54     while(t[pre].ch[1]) pre=t[pre].ch[1];
 55     while(t[nxt].ch[0]) nxt=t[nxt].ch[0];
 56     if(!pre && !nxt) return (void)(root=0);
 57     if(!pre) splay(nxt,root),t[root=nxt].fa=0;
 58     else if(!nxt) splay(pre,root),t[root=pre].fa=0;
 59     else splay(pre,0),splay(nxt,pre),t[nxt].ch[0]=0,pushup(nxt),pushup(pre);
 60     t[x].ch[0]=t[x].ch[1]=0,t[x].siz=1;  //不知道为什么这里不写会 T (懒得想咯应该是编号改完可能会改回来的问题吧)
 61 }
 62 inline void push_front(int x){ //插头
 63     if(!root) return (void)(root=x); int fa=root;
 64     while(t[fa].ch[0]) ++t[fa].siz,fa=t[fa].ch[0];
 65     ++t[fa].siz,t[fa].ch[0]=x,t[x].fa=fa,splay(x,0);
 66 }
 67 inline void push_back(int x){  //插尾
 68     if(!root) return (void)(root=x); int fa=root;
 69     while(t[fa].ch[1]) ++t[fa].siz,fa=t[fa].ch[1];
 70     ++t[fa].siz,t[fa].ch[1]=x,t[x].fa=fa,splay(x,0);
 71 }
 72 inline void split(int x,int id){ //拆出节点
 73     int L=t[x].l,R=t[x].r,ls,rs;
 74     if(L==R) return ; //不用拆
 75     if(L==id){  //最左端
 76         mp[R]=rs=++cnt,mp[id]=x;
 77         t[rs].ch[1]=t[x].ch[1];
 78         t[t[rs].ch[1]].fa=rs;
 79         t[x].ch[1]=rs,t[rs].fa=x;
 80         t[rs].l=L+1,t[rs].r=R,t[x].r=L;
 81         pushup(rs),pushup(x);
 82     } else if(R==id){ //最右端
 83         mp[R-1]=ls=++cnt,mp[id]=x;
 84         t[ls].ch[0]=t[x].ch[0];
 85         t[t[ls].ch[0]].fa=ls;
 86         t[x].ch[0]=ls,t[ls].fa=x;
 87         t[ls].l=L,t[ls].r=R-1,t[x].l=R;
 88         pushup(ls),pushup(x);
 89     } else{  //在中间
 90         mp[id]=x,mp[id-1]=ls=++cnt,mp[R]=rs=++cnt;
 91         t[ls].ch[0]=t[x].ch[0],t[rs].ch[1]=t[x].ch[1];
 92         t[t[ls].ch[0]].fa=ls,t[t[rs].ch[1]].fa=rs;
 93         t[x].ch[0]=ls,t[x].ch[1]=rs,t[ls].fa=t[rs].fa=x;
 94         t[x].l=t[x].r=id,t[ls].l=L,t[ls].r=id-1,t[rs].l=id+1,t[rs].r=R;
 95         pushup(ls),pushup(rs),pushup(x);
 96     }
 97 }
 98 signed main() {
 99     n=read(),m=read(),
100     mp[n]=root=newnode(1,n);
101     for(int x,y,pos,opt;m;--m){
102         opt=read();
103         switch(opt){
104             case 1:
105                 x=read()-ans,y=read()-ans;
106                 pos=mp.lower_bound(x)->second; //map 里面找节点编号
107                 split(pos,x),ans=query(pos); //拆出节点查排名
108                 t[pos].l=t[pos].r=y,mp[y]=pos; //修改信息输答案
109                 print(ans); break;
110             case 2:
111                 x=read()-ans, pos=mp.lower_bound(x)->second;
112                 split(pos,x),ans=query(pos),erase(pos); //拆除节点再删除
113                 push_front(pos),print(ans); break; //节点重新加入树
114             case 3:
115                 x=read()-ans, pos=mp.lower_bound(x)->second;
116                 split(pos,x),ans=query(pos),erase(pos);
117                 push_back(pos),print(ans); break;
118             case 4:
119                 x=read()-ans,ans=get_id(x),print(ans); break; //询问编号直输出
120         }
121     } Ot(); return 0;
122 }

FHQ treap

原文地址:https://www.cnblogs.com/Judge/p/9538306.html