【BZOJ1056】[HAOI2008]排名系统（Splay）
题面
题解
(Splay)随便维护一下就好了，至于名字什么的，我懒得手写哈希表了，直接哈希之后拿(map)压。
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<map>
using namespace std;
#define ll long long
#define ull unsigned long long
#define MAX 250500
#define ls (t[x].ch[0])
#define rs (t[x].ch[1])
#define inf 2000000000
const int base=233;
inline int read()
{
int x=0;bool t=false;char ch=getchar();
while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
if(ch=='-')t=true,ch=getchar();
while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
return t?-x:x;
}
map<ull,int> M;
char name[MAX][20],ch[20];
int len[MAX];
struct Node
{
int ch[2],ff;
int v,size;
}t[MAX];
int rt,n,tot;
void pushup(int x){t[x].size=t[ls].size+t[rs].size+1;}
void rotate(int x)
{
int y=t[x].ff,z=t[y].ff;
int k=t[y].ch[1]==x;
if(z)t[z].ch[t[z].ch[1]==y]=x;t[x].ff=z;
t[y].ch[k]=t[x].ch[k^1];t[t[x].ch[k^1]].ff=y;
t[x].ch[k^1]=y;t[y].ff=x;
pushup(y);pushup(x);
}
void Splay(int x,int goal)
{
while(t[x].ff!=goal)
{
int y=t[x].ff,z=t[y].ff;
if(z!=goal)
(t[y].ch[0]==x)^(t[z].ch[0]==y)?rotate(x):rotate(y);
rotate(x);
}
if(!goal)rt=x;
}
void insert(int u,int id)
{
int x=rt;
while(233)
{
if(!t[x].ch[u>t[x].v])
{
t[id].ff=x;t[id].v=u;t[id].size=1;
t[x].ch[u>t[x].v]=id;
Splay(id,0);
break;
}
else x=t[x].ch[u>t[x].v];
}
}
int Kth(int K)
{
int x=rt;
while(233)
{
if(t[ls].size+1==K)return x;
else if(t[ls].size>=K)x=ls;
else K-=t[ls].size+1,x=rs;
}
}
void Del(int x)
{
Splay(x,0);int rk=t[ls].size;
int l=Kth(rk),r=Kth(rk+2);
Splay(l,0);Splay(r,l);t[r].ch[0]=0;
pushup(r);pushup(l);
t[x].ff=t[x].size=t[x].v=0;
}
int getnum(char *s)
{
int l=strlen(s+1),ret=0;
for(int i=1;i<=l;++i)ret=ret*10+s[i]-48;
return ret;
}
ull gethash(char *s)
{
int l=strlen(s+1);ull h=0;
for(int i=1;i<=l;++i)h=h*base+s[i];
return h;
}
void Output(int x,int &sum)
{
if(sum>=10)return;
if(rs)Output(rs,sum);
if(sum>=10)return;
if(x>2)
{
++sum;
for(int i=1;i<len[x];++i)
putchar(name[x][i]);
putchar(' ');
}
if(ls)Output(ls,sum);
}
int main()
{
n=read();
insert(-inf,++tot);insert(inf,++tot);
while(n--)
{
scanf("%s",ch);
if(ch[0]=='+')
{
ull h=gethash(ch);int sco=read(),id;
if(M[h])id=M[h],Del(id),insert(sco,id);
else
{
insert(sco,M[h]=id=++tot);
for(int i=1,l=strlen(ch);i<l;++i)name[tot][i]=ch[i];
len[tot]=strlen(ch);
}
}
else if(ch[1]>='0'&&ch[1]<='9')
{
int rk=getnum(ch);
int l=Kth(tot-rk+1);
Splay(l,0);int sum=0;
Output(t[l].ch[0],sum);puts("");
}
else
{
ull h=gethash(ch);
int x=M[h];Splay(x,0);
printf("%d
",t[rs].size);
}
if(n%200==0)Splay(rand()%tot+1,0);
}
return 0;
}