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 100100using 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;}