codeforces 149E Martian Strings - AC自动机

传送门
题解：模板题

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<climits>
#include<queue>
#define M 110
#define N 100010
#define LEN 1010
#define toi(c) (c-'A'+1)
#define INF (INT_MAX/10-10)
#define SIG 26
using namespace std;
struct ac_node{
    int ch[SIG+1],val,fail;
}t[M*LEN];int rt,node_cnt;
struct edges{
    int to,pre;
}e[M*LEN];int h[M*LEN],etop;
int len[M],pre[M][LEN],suf[M][LEN];
char s[N],T[M][LEN];
inline int add_edge(int u,int v)
{
    return e[++etop].to=v,e[etop].pre=h[u],h[u]=etop;
}
inline int new_node()
{
    int x=++node_cnt;t[x].val=INF;
    memset(t[x].ch,0,sizeof(t[x].ch));
    return x;
}
inline int insert_ac(int rt,char *s,int n)
{
    int x=rt,c=toi(s[1]);
    for(int i=1;i<=n;i++)
    {
        if(!t[x].ch[c]) t[x].ch[c]=new_node();
        x=t[x].ch[c],c=toi(s[i+1]);
    }
    return 0;
}
queue<int> q;
inline int get_fail(int rt)
{
    while(!q.empty()) q.pop();
    for(int i=1;i<=SIG;i++)
    {
        int &y=t[rt].ch[i];
        if(y) t[y].fail=rt,q.push(y),add_edge(rt,y);
        else y=rt;
    }
    while(!q.empty())
    {
        int x=q.front();q.pop();
        for(int i=1;i<=SIG;i++)
        {
            int &y=t[x].ch[i],f=t[x].fail;
            if(y) t[y].fail=t[f].ch[i],q.push(y),add_edge(t[f].ch[i],y);
            else y=t[f].ch[i];
        }
    }
    return 0;
}
inline int find_ac(int rt,char *s,int n)
{
    for(int i=1,x=rt;i<=n;i++)
        x=t[x].ch[toi(s[i])],t[x].val=min(t[x].val,i);
    return 0;
}
int dfs(int x)
{
    for(int i=h[x],y;i;i=e[i].pre)
        dfs(y=e[i].to),t[x].val=min(t[x].val,t[y].val);
    return 0;
}
inline int clear_ac(int rt,int &n)
{
    memset(h,0,sizeof(h)),etop=0;
    for(int i=rt;i<=n;i++)
        memset(t[i].ch,0,sizeof(t[i].ch));
    return n=0,rt=new_node();
}
inline int get_val(int rt,char *s,int n,int *val)
{
    for(int i=1,x=rt;i<=n;i++)
        x=t[x].ch[toi(s[i])],val[i]=t[x].val;
    return 0;
}
inline int solve(int n,int m,int val[M][LEN])
{
    for(int i=1;i<=m;i++)
        insert_ac(rt,T[i],len[i]);
    get_fail(rt),find_ac(rt,s,n),dfs(rt);
    for(int i=1;i<=m;i++)
        get_val(rt,T[i],len[i],val[i]);
    return 0;
}
inline int Reverse(int n,int m)
{
    for(int i=1;i<=n/2;i++) swap(s[i],s[n-i+1]);
    for(int i=1;i<=m;i++)
        for(int j=1;j<=len[i]/2;j++)
            swap(T[i][j],T[i][len[i]-j+1]);
    return 0;
}
int main()
{
    int n,m,ans=0;rt=new_node();
    scanf("%s",s+1),n=strlen(s+1);
    scanf("%d",&m);
    for(int i=1;i<=m;i++)
        scanf("%s",T[i]+1),len[i]=strlen(T[i]+1);
    solve(n,m,pre),clear_ac(rt,node_cnt);
    Reverse(n,m),solve(n,m,suf);
    for(int i=1;i<=m;i++)
        for(int j=1;j<=len[i]/2;j++)
            swap(suf[i][j],suf[i][len[i]-j+1]);
    for(int i=1;i<=m;i++)
    {
        bool f=false;
        for(int j=1;j<len[i];j++)
            if(pre[i][j]<INF&&suf[i][j+1]<INF&&pre[i][j]<n-suf[i][j+1]+1)
            {   f=true;break;   }
        if(f) ans++;
    }
    printf("%d\n",ans);return 0;
}