题目
题意
给 m ( m < 10 ) m(m<10) m(m<10)个长度小于10的模式串,求长度为 n ( n < 1 e 9 ) n(n<1e9) n(n<1e9)的字符串 S S S,且所有模式串不是 S S S子串(连续),字符串 S S S的个数。
思路
考虑模式串小,将模式串做成AC自动机,考虑构造
S
S
S时在Trie上走,每次不能走到串尾,或者后缀是某个模式串的结点。
考虑
d
p
[
i
]
[
j
]
dp[i][j]
dp[i][j],Trie树从
i
i
i结点走到
j
j
j结点只走一步,有几种方案数,那么
d
p
[
i
]
[
j
]
n
dp[i][j]^n
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;
}