【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)));
        }
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
题目描述 有一个 $n$ 个点的棋盘,每个点上有一个数字 $a_i$,你需要从 $(1,1)$ 走到 $(n,n)$,每次只能往右或往下走,每个格子只能经过一次,路径上的数字和为 $S$。定义一个点 $(x,y)$ 的权值为 $a_x+a_y$,求所有满足条件的路径中,所有点的权值和的最小值。 输入格式 第一行一个整数 $n$。 接下来 $n$ 行,每行 $n$ 个整数,表示棋盘上每个点的数字。 输出格式 输出一个整数,表示所有满足条件的路径中,所有点的权值和的最小值。 数据范围 $1\leq n\leq 300$ 输入样例 3 1 2 3 4 5 6 7 8 9 输出样例 25 算法1 (形dp) $O(n^3)$ 我们可以先将所有点的权值求出来,然后将其看作是一个有权值的图,问题就转化为了在这个图中求从 $(1,1)$ 到 $(n,n)$ 的所有路径中,所有点的权值和的最小值。 我们可以使用形dp来解决这个问题,具体来说,我们可以将这个图看作是一棵,每个点的父节点是它的前驱或者后继,然后我们从根节点开始,依次向下遍历,对于每个节点,我们可以考虑它的两个儿子,如果它的两个儿子都被遍历过了,那么我们就可以计算出从它的左儿子到它的右儿子的路径中,所有点的权值和的最小值,然后再将这个值加上当前节点的权值,就可以得到从根节点到当前节点的路径中,所有点的权值和的最小值。 时间复杂度 形dp的时间复杂度是 $O(n^3)$。 C++ 代码 算法2 (动态规划) $O(n^3)$ 我们可以使用动态规划来解决这个问题,具体来说,我们可以定义 $f(i,j,s)$ 表示从 $(1,1)$ 到 $(i,j)$ 的所有路径中,所有点的权值和为 $s$ 的最小值,那么我们就可以得到如下的状态转移方程: $$ f(i,j,s)=\min\{f(i-1,j,s-a_{i,j}),f(i,j-1,s-a_{i,j})\} $$ 其中 $a_{i,j}$ 表示点 $(i,j)$ 的权值。 时间复杂度 动态规划的时间复杂度是 $O(n^3)$。 C++ 代码
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值