计蒜客 阿里巴巴的手机代理商(困难)[字典树]

题意

有四种操作:
1. 插入一个单词,词频为k
2. 删除所有a这个单词
3. 查询以a为后缀的所有单词词频和
4. 将后缀为a的单词全部更改为后缀为b的单词。
5. 查询当前版本-k的版本以a为后缀的所有单词词频和

题解

对于前四个问询,建立一颗后缀字典树,每一个节点记录size与End,分别表示以当前后缀为后缀的所有单词的词频与以当前后缀为完整单词的单词数,End只更新单词的第一个字母所在的节点,size要对每个节点都更新。update操作则将整个树节点移动至目标位置即可。
对于第5个询问,我们可以离线的读入所有操作,做一遍1,2,3,4操作得到当前版本号与第五个操作的版本号,然后重新做一遍所有询问,对于离线查询操作5。

AC代码

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<string>
#include<iostream>
#include<queue>
#define N 100005
using namespace std;
typedef long long ll;
queue<pair<string,ll> >qq[N];
ll size[1000005],tree[1000005][27],End[1000005],tot,root,n,k;
string op[N],a[N],b[N];
ll ans[N],nnum[N];
ll init(ll root)
{
    for(ll i=0;i<27;i++)
        tree[root][i]=-1;
    size[root]=0;
    End[root]=0;
    return root;
}
ll insert(string buf,ll k,ll k2)
{
    int l=buf.length(),now=root;
    for(int i=0;i<l;i++)
    {
        if(tree[now][buf[i]-'a']==-1)tree[now][buf[i]-'a']=init(++tot);
        now=tree[now][buf[i]-'a'];
        size[now]+=k;
    }
    End[now]+=k2;
    return now;
}
ll getnum(string buf,ll flag,ll &now)
{
    now=root;
    int l=buf.length();
    for(int i=0;i<l;i++)
    {
        now=tree[now][buf[i]-'a'];
        if(now==-1)return 0;
    }
    if(!flag)return End[now];
    return size[now];
}
void del(string buf,ll num,ll num2)
{
    int l=buf.length(),now=root;
    for(int i=0;i<l;i++)
    {
        now=tree[now][buf[i]-'a'];
        size[now]-=num;
    }
    End[now]-=num2;
}
int main()
{
    ll T;
    scanf("%lld",&T);
    while(T--)
    {
        memset(ans,-1,sizeof(ans));
        scanf("%lld",&n);
        ll ntot=0,now;
        for(ll i=1;i<=n;i++)
        {
            cin>>op[i]>>a[i];
            reverse(a[i].begin(),a[i].end());
            if(op[i][0]=='i')cin>>nnum[i],ntot++;
            if(op[i][0]=='d')ntot++;
            if(op[i][0]=='u')cin>>b[i],ntot++,reverse(b[i].begin(),b[i].end());
            if(op[i][0]=='v')cin>>nnum[i];
        }
        for(ll k=0;k<2;k++)
        {
            tot=0;
            ntot=0; 
            init(root);
            if(k==1)
                while(!qq[ntot].empty())
                {
                    ans[qq[ntot].front().second]=getnum(qq[ntot].front().first,1,now);
                    qq[ntot].pop(); 
                }
            for(ll j=1;j<=n;j++)
            {
                ll l=a[j].length();
                if(op[j][0]=='i')
                    insert(a[j],nnum[j],nnum[j]),ntot++;
                if(op[j][0]=='d')
                {
                    ll sum=getnum(a[j],0,now);
                    if(sum==0)
                    {
                        ans[j]=-2;
                        continue;
                    }
                    del(a[j],End[now],End[now]);ntot++;
                }
                if(op[j][0]=='q')
                    ans[j]=getnum(a[j],1,now);
                if(op[j][0]=='u')
                {
                    ll l2=b[j].length(),now2,tmp[27];
                    ll num=getnum(a[j],1,now);
                    ll E=End[now];
                    if(num==0)
                    {
                        ans[j]=-2;
                        continue;
                    }
                    if(getnum(b[j],1,now2))
                    {
                        ans[j]=-3;
                        continue;
                    }
                    for(ll i=0;i<27;i++)
                        tmp[i]=tree[now][i],tree[now][i]=-1;
                    now2=insert(b[j],num,E);
                    for(ll i=0;i<27;i++)
                        tree[now2][i]=tmp[i];
                    del(a[j],num,E);ntot++;
                }
                if(k==1)
                    while(!qq[ntot].empty())
                    {
                        ans[qq[ntot].front().second]=getnum(qq[ntot].front().first,1,now);
                        qq[ntot].pop(); 
                    }
                else if(k==0&&op[j][0]=='v')qq[ntot-nnum[j]].push(make_pair(a[j],j));
            }
        }
        for(ll i=1;i<=n;i++)
        {
            if(ans[i]==-2)printf("Empty\n");
            else if(ans[i]==-3)printf("Conflict\n");
            else if(ans[i]!=-1)printf("%lld\n",ans[i]);
        }
    }
}
阅读更多
文章标签: 字典树
个人分类: trie树/字典树
上一篇计蒜客 阿里巴巴的手机代理商(中等)[字典树]
下一篇Codeforces 965E Short Code [字典树+想法]
想对作者说点什么? 我来说一句

2016校赛标程和数据

2016年05月08日 4.97MB 下载

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

关闭
关闭