[THUSC 2016] 补退选 （Trie树）

link

$solution:$

$Trie$树很显然吧，那么如何去处理每次询问。对于$Trie$树的每个节点放一个$vector$表示其若有$v$个人的最小时间。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
#define ll long long
using namespace std;
inline int read(){
    int f=,ans=;char c=getchar();
    while(c<''||c>''){if(c=='-')f=-;c=getchar();}
    while(c>=''&&c<=''){ans=ans*+c-'';c=getchar();}
    return f*ans;
}
const int MAXN=;
vector<int> ve[MAXN];
int ch[MAXN][],tot,Time[MAXN];
struct Trie{
    void insert(char *s,int opt,int k){
        int len=strlen(s),u=;
        for(int i=;i<len;i++){
            int c=s[i]-'a';
            if(!ch[u][c]) ch[u][c]=++tot;
            u=ch[u][c];
            Time[u]+=opt;
            if(Time[u]>ve[u].size()) ve[u].push_back(k);
        }
    }
    int Query(char *s){
        int len=strlen(s),u=;
        for(int i=;i<len;i++){
            int c=s[i]-'a';
            if(!ch[u][c]) return -;
            u=ch[u][c];
        }return u;
    }
}trie;
int n,ans;
char str[];
int main(){
    // freopen("2.in","r",stdin);
     n=read();
     for(int i=;i<=n;i++){
         int opt=read();
         if(opt==){
             scanf("%s",str);
             trie.insert(str,,i);
         }
         if(opt==){
             scanf("%s",str);
             trie.insert(str,-,i);
         }
         if(opt==){
             scanf("%s",str);
             int pos=trie.Query(str);
             ll a=read(),b=read(),c=read();
             ans=abs(ans);
             ll k=(a*ans+b)%c;
             k=(int)k;
             /*printf("k:%d\n",k);
             printf("siz:%d\n",ve[pos][0]);*/
             if(pos==-){ans=-;printf("%d\n",ans);continue;}
             if(ve[pos].size()>k){printf("%d\n",ans=ve[pos][k]);continue;}
             else {ans=-;printf("%d\n",ans);continue;}

         }
     }
}