GAUSS 2014-AC自动机

https://www.oj.swust.edu.cn/problem/show/2790

题意中文不说了

做法：就是标准的AC自动机+矩阵快速幂

POJ - 2778 的弱化版

直接建AC自动机，如果i到j有一条边，直接g[i][j]++；

最后sum(g^n[0][1-k]就可以了

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define cppiofast ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
const int MAXN=26;
const int N=26;
const ll mod=1e9+7;
struct matrix
{
    ll mat[30][30];
    matrix()
    {
        memset(mat,0,sizeof(mat));
    }
};
struct node
{
    int next[MAXN][N],fail[N],endd[N];
    int root,l;
    int newnode()
    {
        l++;
        for(int i=0;i<N;i++) next[l][i]=0;
        endd[l]=fail[l]=0;
        return l;
    }
    void init()
    {
        l=-1;
        root=newnode();
    }
    void Insert(char *s)
    {
        int len=strlen(s);
        int now=root;
        for(int i=0;i<len;i++)
        {
            if(!next[now][s[i]-'a'])
                next[now][s[i]-'a']=newnode();
            now=next[now][s[i]-'a'];
        }
        endd[now]=1;
    }
    void build()
    {
        queue<int>qu;
        int now=root;
        for(int i=0;i<N;i++)
            if(next[root][i]) qu.push(next[root][i]);
        while(!qu.empty())
        {
            now=qu.front(); qu.pop();
            //endd[now]+=endd[fail[now]];
            for(int i=0;i<N;i++)
            {
                if(!next[now][i]) next[now][i]=next[fail[now]][i];
                else fail[next[now][i]]=next[fail[now]][i],qu.push(next[now][i]);
            }
        }
    }
    matrix buildM()
    {
        matrix A;
        for(int i=0;i<=l;i++)
        {
            for(int j=0;j<N;j++)
            {
                if(!endd[i]&&!endd[next[i][j]])
                    A.mat[i][next[i][j]]++;
            }
        }
        return A;
    }
    int getl()
    {
        return l;
    }
}Aho;
int sz;
matrix mul(matrix A,matrix B)
{
    matrix C;
    for(int i=0;i<=sz;i++)
    {
        for(int j=0;j<=sz;j++)
        {
            for(int k=0;k<=sz;k++)
                C.mat[i][j]=(C.mat[i][j]+A.mat[i][k]*B.mat[k][j])%mod;
        }
    }
    return C;
}
matrix quick(matrix A,ll n)
{
    matrix B;
    for(int i=0;i<=sz;i++) B.mat[i][i]=1;
    while(n)
    {
        if(n&1) B=mul(A,B);
        A=mul(A,A);
        n>>=1;
    }
    return B;
}
char s1[25];
int main()
{
    cppiofast;
    int n;
    Aho.init();
    cin>>n;
    cin>>s1;
    Aho.Insert(s1);
    Aho.build();
    matrix A=Aho.buildM();
    sz=Aho.getl();
    A=quick(A,n);
    ll ans=0;
    for(int i=0;i<=sz;i++)
        ans=(ans+A.mat[0][i])%mod;
    cout<<ans<<endl;
    return 0;
}