HDU 3436 题解

题目大意

有一个队列，开始时第i个数是i。

有三种操作。

Top x：把x放到队头。

Query x：查询x是第几个数。

Rank x：查询第x个数。

n<=10^8,m<=10^5,x<=n

我的做法

离散化+线段树

把top和query中的x离散化，把用不到的区间缩成一个点。

记下每个数在线段树中的位置，线段树中每个位置中的第一个数。

top：把原来的数删除，添加一个新数。

query：输出x在线段树中的位置到线段树最后一个位置之间数的个数。

rank：查询第x个数。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<map>
using namespace std;
struct tree
{
int l,r;
int v;
int sum;
tree *ls,*rs;
tree()
{
l=r=sum=0;
ls=rs=0;
v=0;
}
};
tree *top;
map<int,int> a,b;
int left[500010];
int right[500010];
int cnt2=0;
void mt(tree *&p)
{
if(p->l==p->r)
return;
p->sum=p->ls->sum+p->rs->sum;
}
void build(tree *&p,int l,int r)
{
if(!p)
p=new tree;
p->l=l;
p->r=r;
p->v=0;
p->sum=0;
if(p->l==p->r)
return;
int mid=(l+r)>>1;
build(p->ls,l,mid);
build(p->rs,mid+1,r);
}
void insert(tree *&p,int x,int v,int t)
{
if(p->l==p->r)
{
p->sum+=t;
if(t==1)
{
a[v]=p->l;
b[p->l]=v;
}
return;
}
int mid=(p->l+p->r)>>1;
if(x<=mid)
insert(p->ls,x,v,t);
else
insert(p->rs,x,v,t);
mt(p);
}
void insert2(tree *&p,int x,int l,int r)
{
if(p->l==p->r)
{
p->sum+=r-l+1;
p->v=++cnt2;
left[cnt2]=l;
right[cnt2]=r;
return;
}
int mid=(p->l+p->r)>>1;
if(x<=mid)
insert2(p->ls,x,l,r);
else
insert2(p->rs,x,l,r);
mt(p);
}
int query(tree *&p,int l,int r)
{
if(l<=p->l&&r>=p->r)
return p->sum;
int s=0;
int mid=(p->l+p->r)>>1;
if(l<=mid)
s+=query(p->ls,l,r);
if(r>mid)
s+=query(p->rs,l,r);
return s;
}
int rr(tree *&p,int k)
{
if(p->l==p->r)
{
if(!p->v)
return b[p->l];
return left[p->v]+k-1;
}
int s=p->rs->sum;
if(k<=s)
return rr(p->rs,k);
return rr(p->ls,k-s);
}
int d[100010];
int xx[100010];
int e[100010];
int main()
{
int t;
scanf("%d",&t);
top=0;
int tt=0;
while(t--)
{
memset(d,0,sizeof(d));
memset(xx,0,sizeof(xx));
memset(e,0,sizeof(e));
a.clear();
b.clear();
printf("Case %d:\n",++tt);
int n,m;
scanf("%d%d",&n,&m);
build(top,1,3*m);
int i;
int x;
int cnt=0;
int tot=0;
for(i=1;i<=m;i++)
{
char s[10];
scanf("%s%d",s,&xx[i]);
if(s[0]=='T')
d[i]=1;
else if(s[0]=='R')
d[i]=2;
else
d[i]=3;
if(d[i]!=2)
e[++tot]=xx[i];
}
e[++tot]=1;
e[++tot]=n;
sort(e+1,e+tot+1);
tot=unique(e+1,e+tot+1)-e-1;
for(i=tot;i>=1;i--)
{
insert(top,++cnt,e[i],1);
if(i!=1&&e[i]!=e[i-1]+1)
insert2(top,++cnt,e[i-1]+1,e[i]-1);
}
for(i=1;i<=m;i++)
{
x=xx[i];
if(d[i]==1)
{
int v=a[x];
insert(top,v,x,-1);
insert(top,++cnt,x,1);
}
else if(d[i]==2)
printf("%d\n",rr(top,x));
else
printf("%d\n",query(top,a[x],cnt));
}
}
return 0;
}