[BZOJ4896][Thu Summer Camp2016]补退选
试题描述
X是T大的一名老师,每年他都要教授许多学生基础的C++知识。在T大,每个学生在每学期的开学前都需要选课,每次选课一共分为三个阶段:预选,正选,补退选;其中"补退选"阶段最忙碌。在补退选阶段,学生即可以选课,也可以退课。对于X老师来说,在补退选阶段可能发生以下两种事件:
1:一个姓名为S的学生选了他的课(姓名S将出现在X的已选课学生名单中)
2:一个姓名为S的学生退了他的课(姓名S将从X的已选课学生名单中移除)
同时,X老师对于有哪些学生选了他的课非常关心,所以他会不定时的查询已选课学生名单,每次查询的格式如下:最早在哪个事件之后,姓名以S为前缀的学生数量超过了v
X老师看你骨骼惊奇,所以想用这个问题考考你,你当然不会畏惧,所以勇敢的接下了这个任务。
注意1:学生的姓名可能相同,如果有p个姓名相同的学生都选了X老师的课,则他们的姓名将出现在X老师的名单上p次。
注意2:只有已经选了课的学生才会退课,如果姓名为S的学生退课,则在他退课之前X老师的名单上一定有姓名S。
注意3:选课,退课和查询都被定义为"事件","事件"的编号从1开始
输入
第一行包含一个正整数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。注:输入中的所有字符串均只包含小写字母。
输出
对于每个查询事件,输出一行表示该查询答案。
输入示例
61 lcaq3 lca 0 0 51 lcq3 lc 0 1 52 lcaq3 lc 0 1 5
输出示例
133
数据规模及约定
n<=100000,字符串长度 <= 60,输入中的所有字符串只会包含前 10 个小写字母
题解
建立 Trie 树,每个节点开一个 vector,每次插入时如果比历史最大值要大,暴力将路径上所有节点的 vector 都 push_back 一下。所有 vector 中元素个数总和不超过 60n(想一想,为什么),这与节点数是同阶的。
需要注意:
1. “超过”指严格大于
2. 样例超出了数据范围(所有字符串只会包含前 10 个小写字母,这个条件在测试数据里是真的)
3. a * |ANS| + b 会爆 int!!!!!!
#include#include #include #include #include #include #include using namespace std;int read() { int x = 0, f = 1; char c = getchar(); while(!isdigit(c)){ if(c == '-') f = -1; c = getchar(); } while(isdigit(c)){ x = x * 10 + c - '0'; c = getchar(); } return x * f;}#define maxn 100010#define maxnode 6500010#define maxl 65#define LL long longvector Fir[maxnode];int ToT, rt, ch[maxnode][10], siz[maxnode];char Str[maxl];void Add(int id) { int u = rt, n = strlen(Str); for(int i = 0; i < n; i++) { int c = Str[i] - 'a'; if(!ch[u][c]) ch[u][c] = ++ToT; u = ch[u][c]; siz[u]++; if(siz[u] > Fir[u].size()) Fir[u].push_back(id); } return ;}void Del() { int u = rt, n = strlen(Str); for(int i = 0; i < n; i++) { int c = Str[i] - 'a'; u = ch[u][c]; siz[u]--; } return ;}int Que(int v) { int u = rt, n = strlen(Str); for(int i = 0; i < n; i++) { int c = Str[i] - 'a'; u = ch[u][c]; } if(Fir[u].size() < v) return -1; return Fir[u][v-1];}int main() { ToT = rt = 1; int q = read(), lst = 0; for(int i = 1; i <= q; i++) { int tp = read(); scanf("%s", Str); if(tp == 1) Add(i); if(tp == 2) Del(); if(tp == 3) { int a = read(), b = read(), c = read(), v = ((LL)a * abs(lst) + b) % c + 1; printf("%d\n", lst = Que(v)); } } return 0;}