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