传送门1
传送门2
传送门3
写在前面:我就是不用hash!
思路:
听Shallwe说这道题卡内存要用hash,表示不服开始写trie树
比较关键的是建立trie树节点,Splay节点和字符串(名字)间的映射关系
每个trie树节点对应一个字符,并且记录其是否为end。
Splay节点的id记录它在trie树上的end节点
具体操作如下
插入时,用find看一下有没有之前出现过,如果没有就直接插入,如果有,我们就得到了它的Splay节点,先删除再插入就行了。
查询排名时通过trie树的find函数得到名字的Splay节点,然后把它splay到根,左子树大小+1就是它的排名
查询第k小时,找到对应的第k大的Splay节点,映射回trie树,然后一步步往father上跳,记录路径上的字符,最后倒序输出即可
注意:
1.代码中的对trie树的查询操作和插入操作合到一起了
2.对于相同分数“先到先得”的原则,插入时把等于情况归入当前节点小于需要插入的节点情况即可
3.为了便于查询,我们插入时取分数的相反数,这样查询得到的第k小就是要求的第k大
代码:
#include<bits/stdc++.h>
using namespace std;
int cnt=1,n,tot,root,fa[300005],num[300005],trie[300005][26];
struct Splay
{
int fa,ch[2],siz,data,id;
}a[300005];
char s[12],ch,hash[300005];
int find(char s[])
{
int len=strlen(s),now=1;
for (int i=0;i<len;i++)
{
if (!trie[now][s[i]-65])
trie[now][s[i]-65]=++cnt,
hash[cnt]=s[i],
fa[cnt]=now;
now=trie[now][s[i]-65];
}
if (num[now]) return num[now];
else {num[now]=++tot;a[tot].id=now;return 0;}
}
void ct(int x)
{
a[x].siz=a[a[x].ch[0]].siz+a[a[x].ch[1]].siz+1;
}
void made(int id,int x)
{
a[id].data=x;
a[id].ch[0]=a[id].ch[1]=a[id].fa=0;
a[id].siz=1;
}
void rorate(int x,bool mk)
{
int y=a[x].fa;
a[y].ch[!mk]=a[x].ch[mk];
a[a[x].ch[mk]].fa=y;
if (a[y].fa)
{
if (a[a[y].fa].ch[0]==y) a[a[y].fa].ch[0]=x;
else a[a[y].fa].ch[1]=x;
}
a[x].fa=a[y].fa;
a[y].fa=x;
a[x].ch[mk]=y;
ct(y);ct(x);
}
void splay(int x,int goal)
{
int y;
while (a[x].fa!=goal)
{
y=a[x].fa;
if (a[y].fa==goal)
{
if(a[y].ch[0]==x) rorate(x,1);
else rorate(x,0);
}
else if (a[a[y].fa].ch[0]==y)
{
if (a[y].ch[0]==x) rorate(y,1);
else rorate(x,0);
rorate(x,1);
}
else
{
if (a[y].ch[1]==x) rorate(y,0);
else rorate(x,1);
rorate(x,0);
}
}
if(!goal) root=x;
}
void insert(int id,int x)
{
made(id,x);
if (!root) {root=id;return;}
int now=root,f=1;
while (f)
{
a[now].siz++;
if (a[now].data<=x)
{
if (a[now].ch[1]) now=a[now].ch[1];
else f=0,a[now].ch[1]=id,a[id].fa=now;
}
else
{
if (a[now].ch[0]) now=a[now].ch[0];
else f=0,a[now].ch[0]=id,a[id].fa=now;
}
}
splay(id,0);
}
void del(int x)
{
splay(x,0);
if (!a[x].ch[0]&&!a[x].ch[1]) root=0;
else if (!a[x].ch[0]) a[a[x].ch[1]].fa=0,root=a[x].ch[1];
else if (!a[x].ch[1]) a[a[x].ch[0]].fa=0,root=a[x].ch[0];
else
{
int y=a[x].ch[0];
while (a[y].ch[1]) y=a[y].ch[1];
splay(y,x);
a[a[x].ch[0]].ch[1]=a[x].ch[1];
a[a[x].ch[1]].fa=a[x].ch[0];
a[a[x].ch[0]].fa=0;
root=a[x].ch[0];
ct(a[x].ch[0]);
}
}
void Kth(int k)
{
int now=root;
int p,t=0;
while (now)
{
p=a[a[now].ch[0]].siz;
if (k>p&&k<=p+1) break;
if (k<=p) now=a[now].ch[0];
else k-=(p+1),now=a[now].ch[1];
}
splay(now,0);
now=a[now].id;
while (now!=1)
s[++t]=hash[now],now=fa[now];
for (int i=t;i;i--) printf("%c",s[i]);
}
int rank(int x)
{
splay(x,0);
return a[a[x].ch[0]].siz+1;
}
main()
{
scanf("%d",&n);
int x,y;
while (n--)
{
x=0;
ch=getchar();
while (ch!='+'&&ch!='?') ch=getchar();
if (ch=='+')
{
scanf("%s%d",s,&x);
x=-x;
y=find(s);
if (y) del(y),insert(y,x);
else insert(tot,x);
}
else
{
scanf("%s",s);
int len=strlen(s);
if (s[0]>='0'&&s[0]<='9')
{
for (int i=0;i<len;i++)
x=(x<<3)+(x<<1)+s[i]-48;
int y=min(tot,x+9);
for (int i=x;i<=y;i++)
Kth(i),printf("%c"," \n"[i==y]);
}
else
printf("%d\n",rank(find(s)));
}
}
}