【BZOJ1056】[HAOI2008]排名系统（Splay）

【BZOJ1056】[HAOI2008]排名系统（Splay）

题面

BZOJ
洛谷

题解

\(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\n",t[rs].size);
		}
		if(n%200==0)Splay(rand()%tot+1,0);
	}
	return 0;
}