loj2291 「THUSC 2016」补退选

ref

pkusc 快到了，做点 thusc 的题涨涨 rp……

#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
using namespace std;
typedef long long ll;
int n, k, ch[6000005][10], cnt, len, f[6000005], ans;
char ss[6000005];
vector<int> vec[6000005];
int main(){
	cin>>n;
	for(int j=1; j<=n; j++){
		scanf("%d %s", &k, ss);
		len = strlen(ss);
		if(k==1){
			int u=0;
			for(int i=0; i<len; i++){
				int c=ss[i]-'a';
				if(!ch[u][c])	ch[u][c] = ++cnt;
				u = ch[u][c];
				f[u]++;
				if(f[u]>vec[u].size())	vec[u].push_back(j);
			}
		}
		if(k==2){
			int u=0;
			for(int i=0; i<len; i++){
				int c=ss[i]-'a';
				u = ch[u][c];
				f[u]--;
			}
		}
		if(k==3){
			int a, b, c, u=0;
			if(ans<0)	ans *= -1;
			scanf("%d %d %d", &a, &b, &c);
			int x=((ll)a*ans%c+b)%c;
			ans = 0;
			for(int i=0; i<len; i++){
				c = ss[i] - 'a';
				if(!ch[u][c]){
					ans = -1;
					break;
				}
				u = ch[u][c];
			}
			if(ans!=-1 && x<vec[u].size())
				ans = vec[u][x];
			else	ans = -1;
			printf("%d\n", ans);
		}
	}
	return 0;
}