bzoj 4896 补退选

Posted on | Views:

$Trie$ 树 + $vector$ 暴力维护答案.

把串全部扔进 $Trie$ 树里面,对于 $Trie$ 树上的每个节点,开一个 $vector$ 维护出现过的值的答案.

即,若某个前缀出现次数最大为 $mx$ ,就只维护 $1\sim mx$ 的答案.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
inline int read()
{
	int out=0,fh=1;
	char jp=getchar();
	while ((jp>'9'||jp<'0')&&jp!='-')
		jp=getchar();
	if (jp=='-')
		fh=-1,jp=getchar();
	while (jp>='0'&&jp<='9')
		out=out*10+jp-'0',jp=getchar();
	return out*fh;
}
const int MAXN=1e5+1,L=60,S=10;
char buf[L+10];
int tid=0,len;
struct Trie
{
	int idx;
	Trie(){idx=0;}
	struct node
	{
		int mx,cur,ch[S];
		vector<int> v;
		node(){mx=cur=0;v.push_back(0);}
	}Tree[MAXN*L];
#define root Tree[o]
	void ins(int val)
	{
		int o=0;
		for(int i=0;i<len;++i)
		{
			int c=buf[i]-'a';
			if(!root.ch[c])
				root.ch[c]=++idx;
			o=root.ch[c];
			root.cur+=val;
			if(root.cur>root.mx)
			{
				root.mx=root.cur;
				(root.v).push_back(tid);
			}
		}
	}
	int query(int x)
	{
		int o=0;
		for(int i=0;i<len;++i)
		{
			int c=buf[i]-'a';
			if(!root.ch[c])
				return -1;
			o=root.ch[c];
		}
		if(x>root.mx)
			return -1;
		return root.v[x];
	}
}T;
int lastans=0;
int main()
{
	int m=read();
	while(m--)
	{
		++tid;
		int op=read();
		scanf("%s",buf);
		len=strlen(buf);
		if(op==1)
			T.ins(1);
		else if(op==2)
			T.ins(-1);
		else
		{
			int a=read(),b=read(),c=read();
			int x=(1LL*a*abs(lastans)%c+b)%c+1;
			lastans=T.query(x);
			printf("%d\n",lastans);
		}
	}
	return 0;
}