[Trie] BZOJ4896: [Thu Summer Camp2016]补退选

题意

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开始。
n<=100000，字符串长度 <= 60，输入中的所有字符串只会包含前 10 个小写字母。
强制在线。

题解

看到前缀容易想到Trie，正常的插入删除，并在每个点上开一个vector记录该前缀出现次数递增变化的时刻序列，询问时在上面二分即可。

#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
inline char gc(){
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int getint(){
    char ch=gc(); int res=0,ff=1;
    while(!('0'<=ch&&ch<='9')){ if(ch=='-') ff=-1; ch=gc(); }
    while('0'<=ch&&ch<='9') res=(res<<3)+(res<<1)+ch-'0', ch=gc();
    return res*ff;
}
struct node{
    int cnt; vector< pair<int,int> > evt;
    node* ch[10];
    node(node* son=NULL){ cnt=0; evt.clear(); for(int i=0;i<=9;i++) ch[i]=son; }    
} nil, *null=&nil, *root=null;
typedef node* P_node;
int Q; long long lstans;
void Insert(P_node &p,char* now,int k){
    if(p==null) p=new node(null);
    p->cnt++; 
    int len=p->evt.size();
    if(len==0||p->evt[len-1].first<p->cnt) p->evt.push_back(make_pair(p->cnt,k));
    if((*now)=='\000') return; Insert(p->ch[(*now)-'a'],now+1,k);
}
void Erase(P_node p,char* now){
    p->cnt--; 
    if((*now)=='\000') return;
    Erase(p->ch[(*now)-'a'],now+1);
}
int Query(P_node p,char* now,int val){
    if(p==null) return -1;
    if((*now)=='\000'){
        int len=p->evt.size();
        if(len==0||p->evt[len-1].first<val) return -1;
        int L=0,R=len-1;
        while(L<=R){
            int mid=(L+R)>>1;
            if(p->evt[mid].first==val) return p->evt[mid].second;
            if(p->evt[mid].first<val) L=mid+1;
                                 else R=mid-1;
        }
    }
    return Query(p->ch[(*now)-'a'],now+1,val);
}
char st[65];
int main(){
    freopen("bzoj4896.in","r",stdin);
    freopen("bzoj4896.out","w",stdout);
    Q=getint();
    for(int ii=1;ii<=Q;ii++){
        int pd=getint(),m=-1;
        char ch=gc(); while(!('a'<=ch&&ch<='z')) ch=gc();
        for(;'a'<=ch&&ch<='z';ch=gc()) st[++m]=ch; st[m+1]='\000';
        if(pd==1) Insert(root,st,ii); else
        if(pd==2) Erase(root,st); else
        if(pd==3){
            int t1=getint(),t2=getint(),t3=getint();
            if(lstans==-1) lstans=1;
            printf("%lld\n",lstans=Query(root,st,(lstans*t1+t2)%t3+1));
        }
    }
    return 0;
}