[BZOJ4896][THUSC2016]补退选(Trie)

4896: [Thu Summer Camp2016]补退选

Time Limit: 20 Sec  Memory Limit: 512 MB
Submit: 653  Solved: 228
[Submit][Status][Discuss]

Description

X是T大的一名老师，每年他都要教授许多学生基础的C++知识。在T大，每个学生在每学期的开学前都需要选课，每

次选课一共分为三个阶段：预选，正选，补退选；其中"补退选"阶段最忙碌。在补退选阶段，学生即可以选课，也

可以退课。对于X老师来说，在补退选阶段可能发生以下两种事件：

1:一个姓名为S的学生选了他的课（姓名S将出现在X的已选课学生名单中）

2:一个姓名为S的学生退了他的课（姓名S将从X的已选课学生名单中移除）

同时，X老师对于有哪些学生选了他的课非常关心，所以他会不定时的查询已选课学生名单，每次查询的格式如下

：最早在哪个事件之后，姓名以S为前缀的学生数量超过了vX老师看你骨骼惊奇，所以想用这个问题考考你，你当

然不会畏惧，所以勇敢的接下了这个任务。

注意1：学生的姓名可能相同，如果有p个姓名相同的学生都选了X老师的课，则他们的姓名将出现在X老师的名单上p次。

注意2：只有已经选了课的学生才会退课，如果姓名为S的学生退课，则在他退课之前X老师的名单上一定有姓名S。

注意3：选课，退课和查询都被定义为"事件"，"事件"的编号从1开始

Input

第一行包含一个正整数n，表示一共发生了n个事件。

接下来n行，每行描述一个事件；每行第一个正整数k表示事件类型：

1.如果k=1，表示选课事件，接下来一个字符串S，表示一个姓名为S的学生选了X老师的课

2.如果k=2，表示退课事件，接下来一个字符串S，表示一个姓名为S的学生退了X老师的课

3.如果k=3，表示查询事件，接下来一个字符串S以及三个非负整数a,b,c，表示X老师想知道最早在第几个事件之后

，姓名以S为前缀的学生数量超过了(a*|ANS|+b)%c，|ANS|表示上次查询事件的答案的绝对值，如果当前是第一次

查询，则|ANS|=0；如果任何时刻都没有超过该值，则答案为-1。注：输入中的所有字符串均只包含小写字母。

Output

对于每个查询事件，输出一行表示该查询答案。

Sample Input

6
1 lcaq
3 lca 0 0 5
1 lcq
3 lc 0 1 5
2 lcaq
3 lc 0 1 5

Sample Output

1
3
3

HINT

n<=100000，字符串长度 <= 60，输入中的所有字符串只会包含前 10 个小写字母

Source

鸣谢Sengxian上传

[Submit][Status][Discuss]

直接上Trie，每个点开vector记录超过某个值的最早时间，时空复杂度都与节点数只相差一个常数60。（但是不知道Pascal选手没有vector该怎么做）。

然后就没什么了，注意绕开Discuss里的几个令人无语的问题即可。

#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#define rep(i,l,r) for (int i=l; i<=r; i++)
typedef long long ll;
using namespace std;

const int N=100010;
int n,now,op,x,len,S,cnt,ans,ch[N*60][10],c[N*60];
ll A,B,C;
char s[N];
vector<int>w[N*60];

int main(){
    freopen("bzoj4896.in","r",stdin);
    freopen("bzoj4896.out","w",stdout);
    scanf("%d",&n); cnt=1;
    rep(tim,1,n){
        scanf("%d",&op); scanf("%s",s); x=1; len=strlen(s);
        if (op==1)
            for (int i=0; i<len; i++){
                now=s[i]-'a'; if (!ch[x][now]) ch[x][now]=++cnt;
                x=ch[x][now]; c[x]++;
                if (c[x]>(int)w[x].size()) w[x].push_back(tim);
            }
        else if (op==2)
            for (int i=0; i<len; i++) now=s[i]-'a',x=ch[x][now],c[x]--;
        else{
            scanf("%lld%lld%lld",&A,&B,&C); ans=abs(ans);
            int p=(1ll*ans*A%C+B)%C; ans=0;
            for (int i=0; i<len; i++){
                now=s[i]-'a'; if (!ch[x][now]){ ans=-1; break; } x=ch[x][now];
            }
            if (ans!=-1){ if ((int)w[x].size()<=p) ans=-1; else ans=w[x][p]; }
            printf("%d\n",ans);
        }
    }
    return 0;
}