4896: [Thu Summer Camp2016]补退选

4896: [Thu Summer Camp2016]补退选

Time Limit: 20 Sec Memory Limit: 512 MB
Submit: 315 Solved: 97
[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$
$trie$上每个节点开一个$vector$
表示这个前缀第一次到达 i <script type="math/tex" id="MathJax-Element-119">i</script>次时是在哪一步
剩下的暴力搞搞就行了

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<cmath>
#include<algorithm>
using namespace std;

const int T = 60;
const int maxn = 1E5 + 10;

int n,cnt,LastAns,siz[maxn*T],last[maxn*T],pre[maxn*T],id[maxn*T];
char s[233];

vector <int> v[maxn*T];

inline void Modify(int I,int k)
{
    int len = strlen(s + 1),x = 1;
    for (int i = 1; i <= len; i++)
    {
        int Nex = s[i] - 'a'; bool flag = 0;
        for (int j = last[x]; j; j = pre[j])
            if (id[j] == Nex) {flag = 1; x = j; break;}
        if (!flag)
        {
            id[++cnt] = Nex; pre[cnt] = last[x];
            last[x] = cnt; x = cnt;
        }
        siz[x] += k; if (v[x].size() < siz[x]) v[x].push_back(I);
    }
}

inline void Query(int a,int b,int c)
{
    int len = strlen(s + 1),x = 1;
    for (int i = 1; i <= len; i++)
    {
        int Nex = s[i] - 'a';
        for (int j = last[x]; j; j = pre[j])
            if (id[j] == Nex) {x = j; break;}
    }
    if (!x) {printf("%d\n",LastAns = -1); return;}
    int k = (1LL * a * abs(LastAns) + b) % c;
    if (v[x].size() <= k) {printf("%d\n",LastAns = -1); return;}
    printf("%d\n",LastAns = v[x][k]);
}

int main()
{
    #ifdef DMC
        freopen("DMC.txt","r",stdin);
    #endif

    cin >> n;
    for (int i = 1; i <= n; i++)
    {
        int com,a,b,c;
        scanf("%d%s",&com,s + 1);
        if (com == 1) a = 1;
        else if (com == 2) a = -1;
        else if (com == 3) scanf("%d%d%d",&a,&b,&c);
        if (com <= 2) Modify(i,a); else Query(a,b,c);
    }
    return 0;
}