【Trie+vector】BZOJ4896(Thu Summer Camp2016)[补退选]题解

题目概述

有n个事件，分为三类：
1.插入字符串S。
2.删除字符串S。
3.求最小的pos，使第pos个事件结束后，以S为前缀的字符串个数>v。

解题报告

这种插入和删除字符串还要问前缀（而且还有强制在线）的题目我们不难想到用Trie解决。只是这个询问比较奇怪，但用STL乱搞就很容易了（滑稽:P）。

为每一个Trie节点记录一个vector，表示w分别为1,2,3,…时的最早事件编号，那么每次询问v时输出w为v+1时的最早事件编号就行了（因为每次w只会+1，所以v+1的最早事件编号绝对比>v+1的最早事件编号小），非常暴力……

但是这样vector会不会炸空间呢？不会。思考一下我们会发现只有插入的时候vector才会进入新的元素，删除是不会的（因为删除让w减小了）。而每次插入最多遍历到len（len是字符串长度）个节点，所以所有节点vector元素的总数为n*len，是可以承受的。

ps：这道题题目描述自相矛盾，没有说清楚到底是26个小写字母还是10个小写字母，不过好像用指针动态处理就没什么事了（但我还是开了10=_=）

示例程序

#include<cstdio>
#include<vector>
using namespace std;
typedef long long LL;
const int maxn=100000,maxl=60,maxt=maxn*maxl,maxi=10;

int te,ans;
char s[maxl+5];
struct Trie
{
    struct node
    {
        int w,id;node* son[maxi];vector<int> que;
        node(int a,int b) {w=a;id=b;for (int i=0;i<maxi;i++) son[i]=0;}
    };
    typedef node* P_node;
    int si;P_node ro;
    Trie() {si=0;ro=new node(0,0);}
    int ID(char ch) {return ch-'a';}
    void Insert(char *s,int te)
    {
        P_node pos=ro;
        for (int i=1;s[i];i++)
        {
            if (!pos->son[ID(s[i])]) pos->son[ID(s[i])]=new node(0,++si);
            pos=pos->son[ID(s[i])];pos->w++;
            int num=pos->que.size();
            if (!num||pos->w>num) pos->que.push_back(te);
            //新的w出现了，把事件编号加入vector
        }
    }
    void Delete(char *s)
    {
        P_node pos=ro;
        for (int i=1;s[i];i++) pos=pos->son[ID(s[i])],pos->w--;
    }
    int Find(char *s,int v)
    {
        P_node pos=ro;
        for (int i=1;s[i];i++)
            if (!pos->son[ID(s[i])]) return -1; else
            pos=pos->son[ID(s[i])];
        int num=pos->que.size();
        if (num<=v) return -1;
        return pos->que[v]; //输出v+1的最小事件编号
    }
};
Trie tr;

char readc()
{
    static char buf[100000],*l=buf,*r=buf;
    if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
    if (l==r) return EOF; else return *l++;
}
bool Eoln(char ch) {return ch==10||ch==13||ch==EOF;}
int reads(char *s)
{
    int len=0;char ch=readc();if (ch==EOF) return EOF;
    s[++len]=ch;while ('a'<=s[len]&&s[len]<='z') s[++len]=readc();
    s[len--]=0;return len;
}
int readi(int &x)
{
    int tot=0,f=1;char ch=readc(),lst=ch;
    while ('9'<ch||ch<'0') {if (ch==EOF) return EOF;lst=ch;ch=readc();}
    if (lst=='-') f=-f;
    while ('0'<=ch&&ch<='9') tot=tot*10+ch-48,ch=readc();
    x=tot*f;
    return Eoln(ch);
}
int absi(int x) {if (x<0) return -x; else return x;}
int main()
{
    freopen("program.in","r",stdin);
    freopen("program.out","w",stdout);
    readi(te);
    for (int i=1;i<=te;i++)
    {
        int td;readi(td);reads(s);
        if (td==1) tr.Insert(s,i); else
        if (td==2) tr.Delete(s); else
        {
            int a,b,c;readi(a);readi(b);readi(c);
            printf("%d\n",ans=tr.Find(s,((LL)a*absi(ans)+b)%c));
        }
    }
    return 0;
}