bzoj4896 补退选
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
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
3
3
HINT
n<=100000,字符串长度 <= 60,输入中的所有字符串只会包含前 10 个小写字母
用动态trie维护各个版本的trie,每个节点维护当前前缀出现次数和最大值
查询时二分
1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include<algorithm> 5 #include<cmath> 6 using namespace std; 7 int ch[60*100005][10],Max[60*100005],val[60*100005],root[100001],pos,ans; 8 char s[101]; 9 void insert(int x,int &y,int v) 10 {int i,a,b,j,now; 11 int len=strlen(s); 12 y=++pos; 13 val[y]=val[x]+v; 14 Max[y]=max(Max[x],val[y]); 15 now=y; 16 for (i=0;i<len;i++) 17 { 18 for (j=0;j<10;j++) 19 ch[now][j]=ch[x][j]; 20 a=ch[x][s[i]-'a']; 21 b=ch[now][s[i]-'a']=++pos; 22 val[b]=val[a]+v;Max[b]=max(Max[a],val[b]); 23 x=a;now=b; 24 } 25 } 26 int query(int rt) 27 {int i,len; 28 len=strlen(s); 29 for (i=0;i<len;i++) 30 rt=ch[rt][s[i]-'a']; 31 return Max[rt]; 32 } 33 int find(int id,int lim) 34 { 35 int l=1,r=id,mid,as=-1; 36 while (l<=r) 37 { 38 mid=(l+r)/2; 39 if (query(root[mid])>lim) as=mid,r=mid-1; 40 else l=mid+1; 41 } 42 return as; 43 } 44 int main() 45 {int i,q,opt,a,b,c,x; 46 cin>>q; 47 for (i=1;i<=q;i++) 48 { 49 scanf("%d",&opt); 50 if (opt<=2) 51 { 52 scanf("%s",s); 53 if (opt==1) 54 insert(root[i-1],root[i],1); 55 else insert(root[i-1],root[i],-1); 56 } 57 else 58 { 59 scanf("%s",s); 60 scanf("%d%d%d",&a,&b,&c); 61 x=(1ll*a*abs(ans)+b)%c; 62 root[i]=root[i-1]; 63 ans=find(i,x); 64 printf("%d\n",ans); 65 } 66 } 67 }