[BZOJ4896][Thu Summer Camp2016]补退选

[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。注：输入中的所有字符串均只包含小写字母。

输出

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

输入示例

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

输出示例

1
3
3

数据规模及约定

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

题解

建立 Trie 树，每个节点开一个 vector，每次插入时如果比历史最大值要大，暴力将路径上所有节点的 vector 都 push_back 一下。所有 vector 中元素个数总和不超过 60n（想一想，为什么），这与节点数是同阶的。

需要注意：

1. “超过”指严格大于

2. 样例超出了数据范围（所有字符串只会包含前 10 个小写字母，这个条件在测试数据里是真的）

3. a * |ANS| + b 会爆 int！！！！！！

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cctype>
#include <algorithm>
#include <vector>
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 long

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