【ZOJ 3228】Searching the String 【AC自动机】

题意

给出n个模式串和一个文本串，输出各个模式串在文本串中出现的次数。模式串有两种类型，0类型代表可以有重叠，1类型代表不能有重叠。模式串可能出现重复。

分析

算是AC自动机的模板题？

因为模式串可以重复，所以如果直接插入并且用val数组来保存模式串的编号的话，后面出现的会把前面出现的给覆盖。所以我这里用了一个map来保存每个模式串在trie中的编号。

如何处理1类型不能有重叠的情况？对于1类型的每个模式串，记录一下它的长度和上次匹配到的位置。当再次匹配到这个模式串的时候，看一下这次的位置和上次位置的差有没有大于它的长度，如果大于，则说明这个可以选择不会重叠。

我们在插入模式串的时候不区分是哪种类型，在进行find的时候也不进行区分，找到一个模式串以后，同时更新两种类型。只在最后输出的时候区分一下。

 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 #include <iostream>
 #include <queue>
 #include <map>

 using namespace std;
 const int maxnode=;
 const int sigma_size=;
 const int maxs=+;
 char T[maxs],P[maxs][],type[maxs];
 map<string,int>ms;
 int kase;
 struct AC_Automata{
     int ch[maxnode][sigma_size],val[maxnode],cnt[maxnode][],tim[maxnode];
     int f[maxnode],last[maxnode],len[maxnode];
     int sz;
     void init(){
         sz=;
         memset(ch[],,sizeof(ch[]));
         memset(cnt,,sizeof(cnt));
         memset(tim,,sizeof(tim));
         val[]=;
         ms.clear();
     }
     void insert(char *s){
         int n=strlen(s),u=;
         for(int i=;i<n;i++){
             int c=s[i]-'a';
             if(!ch[u][c]){
                 ch[u][c]=sz;
                 memset(ch[sz],,sizeof(ch[sz]));
                 val[sz++]=;
             }
             u=ch[u][c];
         }
         val[u]=;
         len[u]=n;
         string S=(string)s;
         ms[S]=u;
     }
     void getFail(){
         queue<int>q;
         f[]=last[]=;
         for(int i=;i<sigma_size;i++){
             int u=ch[][i];
             if(u){
                 q.push(u);
                 f[u]=last[u]=;
             }
         }
         while(!q.empty()){
             int r=q.front();q.pop();
             for(int i=;i<sigma_size;i++){
                 int u=ch[r][i];
                 if(!u)continue;
                 q.push(u);
                 int v=f[r];
                 while(v&&!ch[v][i])v=f[v];
                 f[u]=ch[v][i];
                 last[u]=val[f[u]]?f[u]:last[f[u]];
             }
         }
     }
     void print(int i,int pos){
         if(val[i]){
             cnt[i][]++;
             if(tim[i]+len[i]<=pos){
                 cnt[i][]++;
                 tim[i]=pos;
             }
             print(last[i],pos);
         }
     }
     void find(char *s){
         int n=strlen(s),j=;
         for(int i=;i<n;i++){
             int c=s[i]-'a';
             while(j&&!ch[j][c])j=f[j];
             j=ch[j][c];
             if(val[j])
                 print(j,i+);
             else if(last[j])
                 print(last[j],i+);
         }
     }
 }ac;
 int n;
 int main(){
     kase=;
     while(scanf("%s",T)!=EOF){
         ++kase;
         scanf("%d",&n);
         ac.init();
         for(int i=;i<=n;i++){
             scanf("%d %s",&type[i],P[i]);
             ac.insert(P[i]);
         }
         ac.getFail();
         ac.find(T);
         printf("Case %d\n",kase);
         for(int i=;i<=n;i++){
             string S=(string)P[i];
             int u=ms[S];
             printf("%d\n",ac.cnt[u][type[i]]);
           // printf("%d\n",u);
         }
         printf("\n");
     }
 return ;
 }