BZOJ[4896] Trie

$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开始

$Solution$

这就是Manchery大神当初在THU切的题啊！
只需要维护一个$Trie$树，每个节点开一个$vector$维护以其为前缀最早有几个字符串的时间。删除其实并不需要删除。

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

const int N = 101010;
const int M = 62;

inline char get(void) {
    static char buf[100000], *S = buf, *T = buf;
    if (S == T) {
        T = (S = buf) + fread(buf, 1, 100000, stdin);
        if (S == T) return EOF;
    }
    return *S++;
}
inline void read(int &x) {
    static char c; x = 0; int sgn = 0;
    for (c = get(); c < '0' || c > '9'; c = get()) if (c == '-') sgn = 1;
    for (; c >= '0' && c <= '9'; c = get()) x = x * 10 + c - '0';
    if (sgn) x = -x;
}
inline int read(char *s) {
    static char c; int len = 0;
    for (c = get(); c < 'a' || c > 'z'; c = get());
    for (; c >= 'a' && c <= 'z'; c = get()) s[len++] = c;
    s[len] = 0; return len;
}

int n, k, Tcnt, rt, a, b, c, ans;
char s[M];
int ch[N * M][12];
int cnt[N * M];
vector<int> rk[N * M];

inline void NewNode(int &x) {
    if (!x) x = ++Tcnt;
}
inline int Abs(int x) {
    return x < 0 ? -x : x;
}
void Insert(int o, char* s, int t) {
    int len = strlen(s), key;
    for (int i = 0; i < len; i++) {
        key = s[i] - 'a';
        NewNode(ch[o][key]);
        o = ch[o][key];
        if ((++cnt[o]) > rk[o].size()) rk[o].push_back(t);
    }
}
void Delete(int o, char* s) {
    int len = strlen(s), key;
    for (int i = 0; i < len; i++) {
        key = s[i] - 'a';
        o = ch[o][key];
        cnt[o]--;
    }
}
int Query(int o, char *s, int k) {
    int len = strlen(s), key;
    for (int i = 0; i < len; i++) {
        key = s[i] - 'a';
        o = ch[o][key];
    }
    if (rk[o].size() < k) return -1;
    return rk[o][k - 1];
}

int main(void) {
    read(n); NewNode(rt);
    for (int i = 1; i <= n; i++) {
        read(k); read(s);
        if (k == 1) {
            Insert(rt, s, i);
        } else if (k == 2) {
            Delete(rt, s);
        } else if (k == 3) {
            read(a); read(b); read(c);
            a = ((long long)a * Abs(ans) % c + b + c) % c + 1;
            ans = Query(rt, s, a);
            printf("%d\n", ans); 
        }
    }
    return 0;
}