ZOJ 3228 Searching the String(AC自动机)

题意：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3228

思路：AC自动机，不过匹配时要要分可重复使用相同串，和不可重复。。就是再find后，要进行一个初始位置与长度的判断，不能重复使用的，就找到后，更新最小初始位置，

详见注释，，（直接用饶齐的代码分析了，实在不想自己敲，现在也不是很有做题的心情。。）

代码：

#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int maxnode=500100;
const int sigma_size=26;
int ans=0;
struct AC_Automata
{
    int ch[maxnode][sigma_size];
    int val[maxnode];   // 每个字符串的结尾结点都有一个非0的val
    int f[maxnode];     // fail函数
    int last[maxnode];  // last[i]=j表j节点表示的单词是i节点单词的后缀，且j节点是单词节点
    int sz;
    int cnt[maxnode][2];
    int time[maxnode],len[maxnode];
    //初始化0号根节点的相关信息
    void init()
    {
        sz=1;
        memset(ch[0],0,sizeof(ch[0]));
        val[0]=last[0]=f[0]=0;
        cnt[0][1]=cnt[1][0]=0;
    }
    //insert负责构造ch与val数组
    //插入字符串,v必须非0表示一个单词节点
    void insert(char *s)///字典树添加第v个字符串，（v非0）
    {
        int n=strlen(s),u=0;
        for(int i=0; i<n; i++)
        {
            int id=s[i]-'a';
            if(ch[u][id]==0)
            {
                ch[u][id]=sz;
                memset(ch[sz],0,sizeof(ch[sz]));///初始化
                val[sz++]=0;
            }
            u=ch[u][id];
        }
        val[u]=1;///是单词结点
        time[u]=0;///单词第一个字符匹配的最小可能位置
        len[u]=n;///单词长度
        cnt[u][1]=cnt[u][0]=0;///计数初始化
    }
    //getFail函数负责构造f和last数组
    void getFail()
    {
        queue<int> q;
        last[0]=f[0]=0;
        for(int i=0; i<sigma_size; i++)
        {
            int u=ch[0][i];
            if(u)
            {
                f[u]=last[u]=0;
                q.push(u);
            }
        }
        while(!q.empty())// 按BFS顺序计算fail
        {
            int r=q.front(); q.pop();
            for(int i=0; i<sigma_size; i++)
            {
                int u=ch[r][i];
                if(u==0)continue;
                q.push(u);
                int v=f[r];
                while(v && ch[v][i]==0) v=f[v];
                f[u]= ch[v][i];///f表示第u个没有匹配上，返回的位置。
                last[u] =  val[f[u]]?f[u]:last[f[u]];///last返回的是u对应的最近的单词结点
            }
        }
    }
    void print(int i,int x)
    {
        if(val[i])
        {
            cnt[i][0]++;///匹配到。可以重复直接加
            if(time[i]+len[i]<=x)///不可以重复情况，没有重复
            {
                time[i]=x;///更新最小的首位置
                cnt[i][1]++;
            }
            print(last[i],x);///上一个单词结点位置
        }

    }
    // 在s中找出 出现了哪几个模板单词
    void find(char *s)
    {
        int n=strlen(s),j=0;
        for(int i=0; i<n; i++)
        {
            int id=s[i]-'a';
            while(j && ch[j][id]==0) j=f[j];///匹配失败向前返
            j=ch[j][id];
            if(val[j]) print(j,i+1);///匹配成功，
            else if(last[j]) print(last[j],i+1);///前面第一个匹配成功的单词
        }
    }
    int find_T(char *s,int type)///找到单词节点位置，输出对应type
    {
        int n=strlen(s),u=0;
        for(int i=0;i<n;i++)
        {
            int id=s[i]-'a';
            u=ch[u][id];
        }
        return cnt[u][type];
    }
};
AC_Automata ac;
char ch[100005][10];
char s[100010];
int ty[100005];
int main()
{
    int T=0;
    int n;

    while(~scanf("%s",s))
    {T++;
        ac.init();
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            scanf("%d %s",&ty[i],ch[i]);
            ac.insert(ch[i]);
        }
        ac.getFail();
        ac.find(s);
        printf("Case %d\n",T);
        for(int i=1;i<=n;i++)
        {
            printf("%d\n",ac.find_T(ch[i],ty[i]));
        }
        printf("\n");
    }
}