NOIP2017赛前模拟 字符串(AC自动机)

题意：
　给你n个单词（字符串T）和一个文章(字符串S)，q个询问，每次修改S中的一个字符，然后输出所有单词在文章中的出现次数。一开始输出没有修改时的答案。

题解：
　先将每个单词建一个AC自动机，在没有修改时统计一个答案；
　然后我们发现，每次修改只会影响[pos-mx+1,pos+mx-1] (mx为最长的单词的长度，pos为当前修改的位置)的答案，其余多出的则不会影响；
　所以每次修改时我们只需要先将原来区间内的答案删去，修改之后在将这段区间放到AC自动机上匹配，加上新的答案;
　另外还有注意普通的AC自动机在匹配时对于一个节点，会一直迭代，跳它的fail指针，统计答案，为了避免超时，我们在一开始的时候就将一个节点所有fail指针跳的点的答案累加，后面匹配就不用迭代了

#include<cstdio>
#include<algorithm>
#include<cmath>
#include<ctime>
#include<iomanip>
#include<iostream>
#include<cstring>
#include<string>
#include<cctype>
using namespace std;
int ans,n,tot,qq,pos,mx,len;
int q[100010],tag[100010],fail[100010],f[100010][27];
char ch,s[200010];bool vis[100010];

inline void Insert(){
    int t=0;
    for(int i=1;i<=len;i++){
        if(!f[t][s[i]-'a']) f[t][s[i]-'a']=++tot;
        t=f[t][s[i]-'a'];
    }
    tag[t]++;
}

inline void getfail(){
    int head=0,tail=1,j;q[1]=0;
    while(head^tail){
        j=q[++head];
        if(j)
          for(int i=0;i<26;i++)
            if(f[j][i])
              fail[f[j][i]]=f[fail[j]][i];
          for(int i=0;i<26;i++)
            if(!f[j][i])
              f[j][i]=f[fail[j]][i];
            else q[++tail]=f[j][i];
    }
    for(int i=1;i<=tail;i++)
    {
        int u=q[i],v=fail[u];
        tag[u]+=tag[v];
    }
}


inline int ac(int st,int end){
    int now=0,tmp,ret=0;
    for(int i=st;i<=end;i++){
        now=f[now][s[i]-'a'];
        ret+=tag[now];
    }
    return ret;
}

int main(){
//  freopen("string.in","r",stdin);
    scanf("%d%d",&n,&qq);
    for(int i=1;i<=n;i++){
        scanf("%s",s+1);
        len=strlen(s+1);
        mx=max(mx,len);
        Insert();
    }
    getfail();
    scanf("%s",s+1);len=strlen(s+1);
    ans=ac(1,len);
    cout<<ans<<endl;
    for(int i=1;i<=qq;i++){
        scanf("%d %c",&pos,&ch);
        int l=pos-mx+1,r=pos+mx-1;
        if(l<0) l=0; if(r>len) r=len;
        ans-=ac(l,r);s[pos]=ch;
        ans+=ac(l,r);
        cout<<ans<<endl;
    }
    return 0;
}

后记:
　由于这是我做的第三道AC自动机的题，所以模板不熟练，考试的时候就爆炸了，所以以后应该多做一些这样的题