BZOJ 4896: [Thu Summer Camp2016]補退選
阿新 • • 發佈:2018-11-04
trie樹+vector+二分
別忘了abs(ans)
#include<cstdio> #include<algorithm> #include<vector> #include<cstring> #include<cmath> #define mp make_pair #define pr pair<int,int> #define sc second using namespace std; int n,cnt,ch[6000005][11],sz[6000005],Max[6000005]; vector<pr> stack[6000005]; char s[65]; int main(){ scanf("%d",&n); int ans=0; for (int ti=1; ti<=n; ti++){ int cas; scanf("%d%s",&cas,s); int len=strlen(s),x=0; if (cas==1){ for (int i=0; i<len; i++){ if (!ch[x][s[i]-'a']) ch[x][s[i]-'a']=++cnt; x=ch[x][s[i]-'a']; sz[x]++; if (sz[x]>Max[x]){ Max[x]=sz[x]; stack[x].push_back(mp(Max[x],ti)); } } } else if (cas==2){ for (int i=0; i<len; i++){ x=ch[x][s[i]-'a']; sz[x]--; } } else{ for (int i=0; i<len; i++) x=ch[x][s[i]-'a']; int a,b,c; scanf("%d%d%d",&a,&b,&c); int lim=(1ll*a*abs(ans)+b)%c; lim++; if (lim>Max[x]) ans=-1; else{ int id=lower_bound(stack[x].begin(),stack[x].end(),mp(lim,0))-stack[x].begin(); ans=stack[x][id].sc; } printf("%d\n",ans); } } return 0; }