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

来源:互联网 发布:手机网络发短信软件 编辑:程序博客网 时间:2024/05/16 08:02

传送门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)));        }    }}
0 0
原创粉丝点击