题目
题解
设 为第i个号码匹配到第j个不吉利数字的方案数。
显然前i个号码的后缀构成了前j个不吉利数字,而 向 转移相当于在后缀后面新加一个字符,可以联想到AC自动机,然而这个只有一个串所以kmp就足够了。
设 为k位后面加一个字母转移到j的方案数,于是:
我们发现k后面加一个字母转移到j可以用kmp实现。
这个式子是线性的,可以用矩阵优化。
#include<bits/stdc++.h>
using namespace std;
int n,m,K,i,j,k,ne[21],ans;
char s[21];
struct M{
int a[21][21];
friend M operator*(M x,M y){
M z;
memset(z.a,0,sizeof(z.a));
for (int i=0;i<m;i++)
for (int j=0;j<m;j++)
for (int k=0;k<m;k++) z.a[i][j]=(z.a[i][j]+x.a[i][k]*y.a[k][j])%K;
return z;
}
friend M operator^(M x,int y){
M z;
memset(z.a,0,sizeof(z.a));
for (int i=0;i<m;i++) z.a[i][i]=1;
for (;y;y>>=1,x=x*x)
if (y&1) z=z*x;
return z;
}
}A,f;
int main(){
scanf("%d%d%d%s",&n,&m,&K,s);
for (i=1;i<m;i++){
while (j && s[i]!=s[j]) j=ne[j-1];
if (s[i]==s[j]) j++;
ne[i]=j;
}
for (i=0;i<m;i++)
for (j='0';j<='9';j++){
k=i;
while (k && s[k]!=j) k=ne[k-1];
if (s[k]==j) k++;
if (k<m) A.a[i][k]++;
}
f.a[0][0]=1;
f=f*(A^n);
for (i=0;i<m;i++) (ans+=f.a[0][i])%=K;
printf("%d",ans);
}