【BZOJ1056&&1862】【codevs1985】排名系统,Splay+trie树

传送门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)));
        }
    }
}
发布了374 篇原创文章 · 获赞 74 · 访问量 25万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 技术黑板 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览