BZOJ 4896 Thu Summer Camp2016 补退选 Trie树

心情不好跑来刷水题……

题目大意：给定一个字符串集，支持插入、删除、查询某个前缀的出现次数最早何时超过给定的值

开个Trie树维护字符串
直接每个节点开个vector记录一下超过$0,1,2,...$时分别是第几次操作就行了

坑点：
1.超过是指大于
2.题目中的“输入中的所有字符串只会包含前$10$个小写字母”是在放屁……
3.$a*|ANS|+b$会爆int

#include <map>
#include <vector>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define M 100100
using namespace std;

int n,m,last_ans;

struct Trie
{
    map<char,Trie*> son;
    int cnt,max_cnt;
    vector<int> ans;
    void* operator new(size_t)
    {
        static Trie mempool[M*60],*C=mempool;
        return C++;
    }
    Trie* Next(char c)
    {
        if(!son[c])
            son[c]=new Trie;
        return son[c];
    }
    Trie* Next(char c) const
    {
        return son.find(c)==son.end()?0x0:son.find(c)->second;
    }
    friend void Insert(Trie *p,char *s,int cnt)
    {
        while(1)
        {
            p->cnt+=cnt;
            if(p->cnt>p->max_cnt)
            {
                p->max_cnt=p->cnt;
                p->ans.push_back(m);
            }
            if(!*s) return ;
            p=p->Next(*s);++s;
        }
    }
    friend int Query(const Trie *p,char *s,int limit)
    {
        while(*s)
        {
            if(!p->Next(*s))
                return -1;
            p=p->Next(*s);++s;
        }
        return p->max_cnt>limit?p->ans[limit]:-1;
    }
}*root=new Trie;

int main()
{
    static char s[66];
    cin>>n;
    for(int i=1,k,a,b,c;i<=n;i++)
    {
        m=i;
        scanf("%d%s",&k,s);
        if(k==1)
            Insert(root,s,1);
        else if(k==2)
            Insert(root,s,-1);
        else
        {
            scanf("%d%d%d",&a,&b,&c);
            printf("%d\n",last_ans=Query(root,s,((long long)a*abs(last_ans)+b)%c));
        }   
    }
    return 0;
}