poj2778 DNA Sequence//AC自动机+矩阵快速幂

题目

题意

给$m(m<10)$个长度小于10的模式串，求长度为$n(n<1e9)$的字符串$S$，且所有模式串不是$S$子串(连续)，字符串$S$的个数。

思路

考虑模式串小，将模式串做成AC自动机，考虑构造$S$时在Trie上走，每次不能走到串尾，或者后缀是某个模式串的结点。
考虑$dp[i][j]$，Trie树从$i$结点走到$j$结点只走一步，有几种方案数，那么$dp[i][j{]}^n$就是最终状态，就可以统计答案了。
怎么确定后缀是某个模式串的结点，可以通过fail树上差分，或者直接构造自动机bfs的时候，前驱如果是串尾，把后继也打上串尾标记。
白书上p368有单模式串的例题hhhh

#include  <map>
#include  <set>
#include  <cmath>
#include  <queue>
#include  <cstdio>
#include  <vector>
#include  <climits>
#include  <cstring>
#include  <cstdlib>
#include  <iostream>
#include  <algorithm>
using namespace std;
#define FI first
#define SE second
#define LL long long
#define MP make_pair
#define PII pair<int,int>
#define SZ(a) (int)a.size()
const double pai = acos(-1);
const double eps = 1e-10;
const LL mod = 1e5;
const int MXN = 1e6+5;
int mp[105][105];
int tr[MXN][4],col[MXN],fail[MXN],nonum;
int id[300];
void ins(char *x){ //插入
    int le=strlen(x),p=0;
    for(int i=0;i<le;i++){
        int c=id[(int)x[i]];
        if(!tr[p][c]) tr[p][c]=++nonum;
        p=tr[p][c];
    }
    col[p]++; //字符串x数量++
}
void ACGet(){
    queue<int>q;
    memset(fail,0,sizeof(fail));
    for(int i=0;i<4;i++){
        if(tr[0][i]) q.push(tr[0][i]);
    }
    while(!q.empty()){
        int now=q.front();q.pop();
        for(int i=0;i<4;i++){
            if(tr[now][i]){
                fail[tr[now][i]]=tr[fail[now]][i];//指向前一个fail(相同最长后缀)
                //fail树上父亲是串尾的话，儿子也不能走到
                if(col[tr[fail[now]][i]]) col[tr[now][i]]=1;
                q.push(tr[now][i]);
            }
            else tr[now][i]=tr[fail[now]][i];
        }
    }
    for(int i=0;i<=nonum;i++){
        if(col[i])continue;
        for(int j=0;j<4;j++){
            if(col[tr[i][j]])continue ;
            mp[i][tr[i][j]]++;
        }
    }
}
/********MatQuickPow********************************************************************/

struct mat{
    LL m[105][105];
};
mat mat_mul(mat aa,mat bb,int nn){
    mat re;memset(re.m,0,sizeof(re.m));
    for(int i=0;i<nn;i++){
        for(int j=0;j<nn;j++){
            for(int k=0;k<nn;k++){
                re.m[i][j]=(re.m[i][j]+aa.m[i][k]*bb.m[k][j])%mod;
            }
        }
    }
    return re;
}
mat mat_quick_mod(mat jin,LL x,int nn){
    mat re;memset(re.m,0,sizeof(re.m));
    for(int i=0;i<nn;i++) re.m[i][i]=1;
    while(x){
        if(x&1) re=mat_mul(re,jin,nn);
        x=x>>1;jin=mat_mul(jin,jin,nn);
    }
    return re;
}/********MatQuickPow********************************************************************/
int main(){
    id['A']=0,id['C']=1,id['T']=2,id['G']=3;
    LL n,m;cin>>m>>n;
    for(int i=0;i<m;i++){
        char ss[15];cin>>ss;ins(ss);
    }
    ACGet();
    mat a;
    for(int i=0;i<=nonum;i++){
        for(int j=0;j<=nonum;j++) a.m[i][j]=mp[i][j];
    }
    mat b=mat_quick_mod(a,n,nonum+1);
    LL ans=0;
    for(int i=0;i<=nonum;i++) ans+=b.m[0][i],ans%=mod;
    cout<<ans;
    return 0;
}