Codoforces 1202E AC自动机

题目：https://codeforces.com/contest/1202/problem/E

题意：

给你一个匹配串t，n个模式串s，任意选两个拼接在一起si+sj，输出所有拼接串在t中出现的次数和

题解：

很简单的ac自动机，然而，还是没想到。。。

根据ac自动机fail数组的特性，一个串fail的父亲一定是这个串的后缀，所以我们可以得到以t串的某个位置为结尾的匹配串的个数；再建一个自动机，输入反的模式串，查询反的匹配串t，我们可以得到以t串的的某个位置为结尾的匹配串的个数，即在原匹配串中，以该位为开头的模式串个数。

最后遍历匹配串t，即枚举字符串si，sj的分割点，求出答案

#include<bits/stdc++.h>
using namespace std;
typedef __int64 ll;
const ll mod=1e9+7;
const int maxn=2e5+10;
struct Tire{
    static const int NODENUM=(int)1e6+5,R=26;
    int nxt[NODENUM][R],fail[NODENUM],ed[NODENUM];
    int rt,tot;
    int dp[NODENUM];
    int newnode(){
        for(int i=0;i<R;i++)nxt[tot][i]=-1;
        ed[tot]=0;
        return tot++;
    }
    void init(){
        tot=0;
        rt=newnode();
    }
    void insert(char *s){
        int now=rt,len=strlen(s);
        for(int i=0;i<len;i++){
            int val=s[i]-'a';
            if(nxt[now][val]==-1)nxt[now][val]=newnode();
            now=nxt[now][val];
        }
        ed[now]++;
    }
    void build(){
        queue<int>q;
        fail[rt]=rt;
        for(int i=0;i<R;i++){
            if(nxt[rt][i]==-1)nxt[rt][i]=rt;
            else {
                fail[nxt[rt][i]]=rt;
                q.push(nxt[rt][i]);
            }
        }
        while(!q.empty()){
            int now=q.front();q.pop();
            ed[now]+=ed[fail[now]];
            for(int i=0;i<R;i++){
                if(nxt[now][i]==-1)nxt[now][i]=nxt[fail[now]][i];
                else {
                    fail[nxt[now][i]]=nxt[fail[now]][i];
                    q.push(nxt[now][i]);
                }
            }
        }
    }
    void query(char *s){
        int now=rt,len=strlen(s);
        for(int i=0;i<len;i++){
            int val=s[i]-'a';
            now=nxt[now][val];
            dp[i]=ed[now];
        }
    }
}AC,ac;
char s[maxn],t[maxn];
int main()
{
    int n;
    scanf("%s %d",t,&n);
    AC.init();
    ac.init();
    while(n--){
        scanf("%s",s);
        AC.insert(s);
        reverse(s,s+strlen(s));
        ac.insert(s);
    }
    AC.build();ac.build();
    AC.query(t);
    int len=strlen(t);
    reverse(t,t+strlen(t));
    ac.query(t);
    ll sum=0;
    for(int i=0;i<len;i++){
        if(len-i-2<0) break;
        sum+=1ll*AC.dp[i]*ac.dp[len-i-2];
    }
    printf("%lld\n",sum);
    return 0;
}