题意:
n个人站成一排 一开始是从1到n有序的 现在有三个操作 Top操作是将一个人排到队首 Query操作是询问某个人现在排第几 Rank操作是询问排某个位置的人是谁
思路:
将队伍扭来扭去… 很像splay的旋转吧(哪像了!!)
这是个不错的splay题…
首先 n很大 但是操作不多 想到离散化
离散化还有个技巧 我们发现只有top和query操作对单人进行 rank和人没什么关系
所以要把top和query操作的人单独拿出来 那么其他的人就可以用区间的形式来表示了
我们只需要开一个num数字来表示树上这个点(表示一个区间)包含几个人
然后就是对每个操作的处理 从简到难
Rank操作最简单 它根本和人没关系 只要找第k个位置就好了 不过我们现在的点表示的是一个区间
那也不难处理 开一个sum数组对num进行维护 就可以简单的找到第k个人所在区间 算一下他是谁就好
接着是Query操作 这个有点烦 因为我们不知道包含k这个人的区间是哪个
其实好解决 开个idx数字记录树上的点表示第几个区间 再开一个qu数组表示某个区间对应树上的点就好
找到了点然后呢? 把它转到根 然后sum[L(root)]+1就是答案!!
最后是Top 有了query的铺垫那么我们只要找到这个点 把它转到根 再把它的前驱转到根下面
删除根节点 再把删掉的这个点插到队首 一个top操作就完成了
代码:
#include<cstdio> #include<cstring> #include<algorithm> #include<iostream> using namespace std; #define keytree ch[ch[root][1]][0] #define L(x) ch[x][0] #define R(x) ch[x][1] #define N 201000 int T,t,n,m,tot,root; int ch[N][2],pre[N],size[N],num[N],sum[N],idx[N],qu[N],a[N],s[N],e[N],op[N],opn[N]; void newnode(int &u,int fa,int w) { u=++tot; L(u)=R(u)=0; pre[u]=fa; size[u]=1; if(w) { sum[u]=num[u]=e[w]-s[w]+1; idx[w]=u; qu[u]=w; } } void up(int u) { size[u]=size[L(u)]+size[R(u)]+1; sum[u]=num[u]+sum[L(u)]+sum[R(u)]; } void down(int u) { } void rotate(int u,int kind) { int fa=pre[u]; down(fa); down(u); ch[fa][!kind]=ch[u][kind]; pre[ch[u][kind]]=fa; if(pre[fa]) ch[pre[fa]][ch[pre[fa]][1]==fa]=u; pre[u]=pre[fa]; ch[u][kind]=fa; pre[fa]=u; up(fa); } void splay(int u,int goal) { int fa,kind; down(u); while(pre[u]!=goal) { if(pre[pre[u]]==goal) rotate(u,L(pre[u])==u); else { fa=pre[u]; kind=L(pre[fa])==fa; if(ch[fa][kind]==u) { rotate(u,!kind); rotate(u,kind); } else { rotate(fa,kind); rotate(u,kind); } } } up(u); if(goal==0) root=u; } int getkth(int u,int k) { down(u); int tmp=size[L(u)]+1; if(tmp==k) return u; if(tmp>k) return getkth(L(u),k); else return getkth(R(u),k-tmp); } int getkperson(int u,int k) { down(u); int tmp=sum[L(u)]; if(k>tmp&&k<=tmp+num[u]) return s[qu[u]]+k-tmp-1; if(tmp>=k) return getkperson(L(u),k); else return getkperson(R(u),k-tmp-num[u]); } void build(int &u,int l,int r,int fa) { if(l>r) return ; int mid=(l+r)>>1; newnode(u,fa,mid); build(L(u),l,mid-1,u); build(R(u),mid+1,r,u); up(u); } void init() { root=tot=0; L(root)=R(root)=pre[root]=size[root]=num[root]=sum[root]=idx[root]=0; newnode(root,0,0); newnode(R(root),root,0); build(keytree,1,n,R(root)); up(R(root)); up(root); } int getpre(int u) { down(u); u=L(u); down(u); while(R(u)) { u=R(u); down(u); } return u; } void remove() { if(L(root)) { int i=getpre(root); splay(i,root); R(i)=R(root); pre[R(root)]=i; root=L(root); pre[root]=0; up(root); } else { root=R(root); pre[root]=0; } } void insert(int u) { keytree=u; pre[u]=R(root); L(u)=R(u)=0; up(u); up(pre[u]); up(root); } int main() { //freopen("test.in","r",stdin); //freopen("test.out","w",stdout); int i,j,k; char str[3]; while(~scanf("%d",&T)) { for(t=1;t<=T;t++) { printf("Case %d:\n",t); scanf("%d%d",&n,&m); for(i=j=1;i<=m;i++) { scanf("%s%d",str,&opn[i]); if(str[0]=='T') { op[i]=1; a[j++]=opn[i]; } else if(str[0]=='Q') { op[i]=2; a[j++]=opn[i]; } else op[i]=3; } sort(a+1,a+j); a[0]=0; for(i=k=1;i<j;i++) { if(a[i]!=a[i-1]) { if(a[i]-a[i-1]>1) { s[k]=a[i-1]+1; e[k]=a[i]-1; k++; } s[k]=a[i]; e[k]=a[i]; k++; } } if(a[j-1]!=n) { s[k]=a[j-1]+1; e[k]=n; } else k--; n=k; init(); for(i=1;i<=m;i++) { if(op[i]==1) { k=lower_bound(s+1,s+n+1,opn[i])-s; k=idx[k]; splay(k,0); remove(); splay(getkth(root,1),0); splay(getkth(root,2),root); insert(k); } else if(op[i]==2) { k=lower_bound(s+1,s+n+1,opn[i])-s; k=idx[k]; splay(k,0); printf("%d\n",sum[L(root)]+1); } else if(op[i]==3) { k=getkperson(root,opn[i]); printf("%d\n",k); } } } } return 0; }
时间: 2024-08-11 14:25:28