首页 > 代码库 > 【bzoj4896】补退选

【bzoj4896】补退选

傻逼题。

每个点维护下vector,然后随便做。

#include<bits/stdc++.h>
const int N=300000;
using namespace std;
typedef long long ll;
struct Node{
    int ch[20],size;
    vector<int> c;
}t[N];
int n,cnt,ans,len;char s[200];
inline int read(){
    int f=1,x=0;char ch;
    do{ch=getchar();if(ch==-)f=-1;}while(ch<0||ch>9);
    do{x=x*10+ch-0;ch=getchar();}while(ch>=0&&ch<=9);
    return f*x; 
}
inline void ins(char *s,int id){
    int now=1;
    for(int i=0;i<len;i++){
        int v=s[i]-a;
        if(!t[now].ch[v])t[now].ch[v]=++cnt;
        now=t[now].ch[v];t[now].size++;
        if(t[now].size>t[now].c.size())t[now].c.push_back(id);
    }
}
inline void del(char *s){
    int now=1;
    for(int i=0;i<len;i++){
        int v=s[i]-a;now=t[now].ch[v];
        t[now].size--;
    }
}
inline int query(char *s){
    int now=1,v;
    for(int i=0;i<len&&now;i++)v=s[i]-a,now=t[now].ch[v];
    int a=read(),b=read(),c=read();
    a=(1LL*a*ans+b)%c;
    if(!now||a>t[now].c.size()){
        ans=1;
        return -1;
    }
    ans=t[now].c[a];
    return ans;
}
int a,b,c,opt;
int main(){
    n=read();cnt=1;
    for(int i=1;i<=n;i++){
        opt=read();scanf("%s",s);len=strlen(s);
        if(opt==1)ins(s,i);
        if(opt==2)del(s);
        if(opt==3)printf("%d\n",query(s));
    }
}

 

【bzoj4896】补退选