1. 程式人生 > >bzoj4896 [Thu Summer Camp2016]補退選

bzoj4896 [Thu Summer Camp2016]補退選

lin 前綴和 a* freopen while sca 可能 getchar() 輸出

Description

X是T大的一名老師,每年他都要教授許多學生基礎的C++知識。在T大,每個學生在每學期的開學前都需要選課,每 次選課一共分為三個階段:預選,正選,補退選;其中"補退選"階段最忙碌。在補退選階段,學生即可以選課,也 可以退課。對於X老師來說,在補退選階段可能發生以下兩種事件: 1:一個姓名為S的學生選了他的課(姓名S將出現在X的已選課學生名單中) 2:一個姓名為S的學生退了他的課(姓名S將從X的已選課學生名單中移除) 同時,X老師對於有哪些學生選了他的課非常關心,所以他會不定時的查詢已選課學生名單,每次查詢的格式如下 :最早在哪個事件之後,姓名以S為前綴的學生數量超過了vX老師看你骨骼驚奇,所以想用這個問題考考你,你當 然不會畏懼,所以勇敢的接下了這個任務。 註意1:學生的姓名可能相同,如果有p個姓名相同的學生都選了X老師的課,則他們的姓名將出現在X老師的名單上p次。 註意2:只有已經選了課的學生才會退課,如果姓名為S的學生退課,則在他退課之前X老師的名單上一定有姓名S。 註意3:選課,退課和查詢都被定義為"事件","事件"的編號從1開始

Input

第一行包含一個正整數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。註:輸入中的所有字符串均只包含小寫字母。

Output

對於每個查詢事件,輸出一行表示該查詢答案。

Sample Input

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

Sample Output

1
3
3

HINT

n<=100000,字符串長度 <= 60,輸入中的所有字符串只會包含前 10 個小寫字母

正解:可持久化$trie$樹+二分。

直接建一個可持久化$trie$樹,維護前綴和與前綴最大值。

查詢的時候二分一下在哪個位置就行了。

 1 #include <bits/stdc++.h>
 2 #define
il inline 3 #define RG register 4 #define ll long long 5 #define N (100010) 6 7 using namespace std; 8 9 int ch[60*N][10],sum[60*N],val[60*N],rt[N],n,sz,len,ans; 10 char s[110]; 11 12 il int gi(){ 13 RG int x=0,q=1; RG char ch=getchar(); 14 while ((ch<0 || ch>9) && ch!=-) ch=getchar(); 15 if (ch==-) q=-1,ch=getchar(); 16 while (ch>=0 && ch<=9) x=x*10+ch-48,ch=getchar(); 17 return q*x; 18 } 19 20 il void insert(RG int x,RG int &y,char *s,RG int len,RG int v){ 21 val[y=++sz]=val[x]+v,sum[y]=max(sum[x],val[y]); RG int now=y; 22 for (RG int i=1,a,b;i<=len;++i){ 23 for (RG int j=0;j<10;++j) ch[now][j]=ch[x][j]; 24 a=ch[x][s[i]-a],b=ch[now][s[i]-a]=++sz; 25 val[b]=val[a]+v,sum[b]=max(sum[a],val[b]),x=a,now=b; 26 } 27 return; 28 } 29 30 il int query(RG int x,char *s,RG int len){ 31 for (RG int i=1;i<=len;++i) x=ch[x][s[i]-a]; 32 return sum[x]; 33 } 34 35 il int find(RG int id,char *s,RG int len,RG int lim){ 36 RG int l=1,r=id,mid,ans=-1; 37 while (l<=r){ 38 mid=(l+r)>>1; 39 if (query(rt[mid],s,len)>lim) ans=mid,r=mid-1; else l=mid+1; 40 } 41 return ans; 42 } 43 44 int main(){ 45 #ifndef ONLINE_JUDGE 46 freopen("trie.in","r",stdin); 47 freopen("trie.out","w",stdout); 48 #endif 49 n=gi(); 50 for (RG int i=1,a,b,c;i<=n;++i){ 51 RG int op=gi(); scanf("%s",s+1),len=strlen(s+1); 52 if (op<=2) insert(rt[i-1],rt[i],s,len,op==1 ? 1 : -1); else{ 53 a=gi(),b=gi(),c=gi(),ans=(1LL*a*abs(ans)+b)%c; 54 rt[i]=rt[i-1],ans=find(i,s,len,ans),printf("%d\n",ans); 55 } 56 } 57 return 0; 58 }

bzoj4896 [Thu Summer Camp2016]補退選