题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=3436
题目思路:splay,也可以用线段树或树状数组,主要难点在离散化,这个离散化是把点和点这间的区间都离散化,离散化后的点要保存原来区间的长度和起始点等信息,这样就可用splay模拟了,top操作可以先删点再插入,也可以将要删的点伸展到根结点,然后将它的左子树接到它后继的左边,这样它便成为整棵树最左边的结点了。由于人的编号和结点的index是一一对应的,不会改变,所以可以把这个关系存起来,这样访问的时候就可以直接通过人的编号找到结点了。这个题我tle了很久,结果才发现最前面的那段区间没存,真是悲剧啊!
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<string>
#include<queue>
#include<algorithm>
#include<vector>
#include<stack>
#include<list>
#include<iostream>
#include<map>
using namespace std;
#define inf 0x3f3f3f3f
#define M 300000
int max(int a,int b)
{
return a>b?a:b;
}
int min(int a,int b)
{
return a<b?a:b;
}
char op[M][10];
int d[M],pos[M],nodenum[M],p[M],ch[M][2],s[M],v[M],begin[M],end[M],num[M];
int root,top1,N;
int change;
struct node
{
int d,id;
bool operator<(const node a)const
{
return d<a.d;
}
}li[M];
void vist(int x)
{
if(!x) return;
printf("x %d p %d ch0 %d ch1 %d s %d v %d\n",x,p[x],ch[x][0],ch[x][1],s[x],v[x]);
vist(ch[x][0]);
vist(ch[x][1]);
}
void debug()
{
puts("debug");
vist(root);
}
void up(int x)
{
int l=ch[x][0],r=ch[x][1];
s[x]=s[l]+s[r]+num[x];
}
void newnode(int &x,int val,int pre)
{
x=++top1;
ch[x][0]=ch[x][1]=0;
p[x]=pre;
v[x]=val;
}
void build(int &x,int l,int r,int pre)
{
if(l>r) return;
int mid=(l+r)>>1;
newnode(x,mid,pre);
nodenum[mid]=x;
num[x]=end[mid]-begin[mid]+1;
s[x]=end[mid]-begin[mid]+1;
build(ch[x][0],l,mid-1,x);
build(ch[x][1],mid+1,r,x);
up(x);
}
void init()
{
top1=0;
p[0]=ch[0][0]=ch[0][1]=s[0]=num[0]=v[0]=0;
newnode(root,inf,0);
s[root]=1;
num[root]=1;
build(ch[root][0],1,N,root);
up(root);
}
void rot(int x,int f)
{
int y=p[x];
p[ch[x][f]]=y;
ch[y][!f]=ch[x][f];
p[x]=p[y];
if(p[y]) ch[p[y]][ch[p[y]][1]==y]=x;
p[y]=x;
ch[x][f]=y;
up(y);
}
void splay(int x,int goal)
{
while(p[x]!=goal)
{
if(p[p[x]]==goal) rot(x,ch[p[x]][0]==x);
else
{
int y=p[x],f=ch[p[y]][0]==y;
if(ch[y][f]==x)
{
rot(x,!f);
}
else
rot(y,f);
rot(x,f);
}
}
up(x);
if(!goal) root=x;
}
void top(int val)
{
int x=nodenum[pos[val]];
splay(x,0);
int t1=ch[root][1];
while(ch[t1][0]) t1=ch[t1][0];
splay(t1,root);
p[ch[root][0]]=t1;
ch[t1][0]=ch[root][0];
ch[root][0]=0;
up(t1);
up(root);
}
void query(int val)
{
int x=nodenum[pos[val]];
splay(x,0);
printf("%d\n",s[ch[root][0]]+num[x]);
}
void rank(int id)
{
int x=root;
while(1)
{
if(s[ch[x][0]]>=id)
x=ch[x][0];
else
{
if(s[ch[x][0]]+num[x]>=id)
break;
id-=s[ch[x][0]]+num[x];
x=ch[x][1];
}
}
int tmp=id-(s[ch[x][0]]);
printf("%d\n",begin[v[x]]+tmp-1);
splay(x,0);
}
int main()
{
int t,n,q,i,count=1;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&q);
int cnt=0;
int ma=0;
for(i=0;i<q;i++)
{
scanf("%s%d",op[i],&d[i]);
if(op[i][0]=='T'||op[i][0]=='Q')
{
li[cnt].d=d[i];
li[cnt++].id=i;
if(op[i][0]=='T')
ma=max(ma,d[i]);
}
}
sort(li,li+cnt);
N=0;
if(li[0].d>1)
{
++N;
begin[N]=1;
end[N]=li[0].d-1;
}
++N;
begin[N]=end[N]=li[0].d;
pos[li[0].id]=N;
for(i=1;i<cnt&&li[i].d<=ma;i++)
{
if(li[i].d==li[i-1].d)
{
pos[li[i].id]=N;
continue;
}
if(li[i].d-li[i-1].d>1)
{
++N;
begin[N]=li[i-1].d+1;
end[N]=li[i].d-1;
}
++N;
begin[N]=end[N]=li[i].d;
pos[li[i].id]=N;
}
init();
printf("Case %d:\n",count++);
for(i=0;i<q;i++)
{
if(op[i][0]=='T')
top(i);
else if(op[i][0]=='Q')
{
if(d[i]>ma)
{
printf("%d\n",d[i]);
continue;
}
query(i);
}
else
{
if(d[i]>ma)
{
printf("%d\n",d[i]);
continue;
}
rank(d[i]);
}
}
}
}