bzoj4896 补退选

Posted 2020-10-29 Z-Y-Y-S

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<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 }