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;
}