【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=700000;
const int sigma_size=26;
const int maxs=100000+100;
char T[maxs],P[maxs][20],type[maxs];
map<string,int>ms;
int kase;
struct AC_Automata{
    int ch[maxnode][sigma_size],val[maxnode],cnt[maxnode][2],tim[maxnode];
    int f[maxnode],last[maxnode],len[maxnode];
    int sz;
    void init(){
        sz=1;
        memset(ch[0],0,sizeof(ch[0]));
        memset(cnt,0,sizeof(cnt));
        memset(tim,0,sizeof(tim));
        val[0]=0;
        ms.clear();
    }
    void insert(char *s){
        int n=strlen(s),u=0;
        for(int i=0;i<n;i++){
            int c=s[i]-'a';
            if(!ch[u][c]){
                ch[u][c]=sz;
                memset(ch[sz],0,sizeof(ch[sz]));
                val[sz++]=0;
            }
            u=ch[u][c];
        }
        val[u]=1;
        len[u]=n;
        string S=(string)s;
        ms[S]=u;
    }
    void getFail(){
        queue<int>q;
        f[0]=last[0]=0;
        for(int i=0;i<sigma_size;i++){
            int u=ch[0][i];
            if(u){
                q.push(u);
                f[u]=last[u]=0;
            }
        }
        while(!q.empty()){
            int r=q.front();q.pop();
            for(int i=0;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][0]++;
            if(tim[i]+len[i]<=pos){
                cnt[i][1]++;
                tim[i]=pos;
            }
            print(last[i],pos);
        }
    }
    void find(char *s){
        int n=strlen(s),j=0;
        for(int i=0;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+1);
            else if(last[j])
                print(last[j],i+1);
        }
    }
}ac;
int n;
int main(){
    kase=0;
    while(scanf("%s",T)!=EOF){
        ++kase;
        scanf("%d",&n);
        ac.init();
        for(int i=1;i<=n;i++){
            scanf("%d %s",&type[i],P[i]);
            ac.insert(P[i]);
        }
        ac.getFail();
        ac.find(T);
        printf("Case %d
",kase);
        for(int i=1;i<=n;i++){
            string S=(string)P[i];
            int u=ms[S];
            printf("%d
",ac.cnt[u][type[i]]);
          // printf("%d
",u);
        }
        printf("
");
    }
return 0;
}