首先看到这道题,考虑DP,最容易想到的状态是令f[i]为准考证为i位数,满足条件的方案数,但是由于这个状态太简单,我们由这个状态知道的信息太少,因此我们考虑换一个状态,看到这道题我们容易想到KMP,因为它要求准考证号的这个字符串中不能含有不吉利数字的字符串,那么我们考虑一种状态为f[i][j],表示准考证为i位数,不吉利的字符串匹配到了j位,满足条件的方案数。那么转移就为f[i][j]-->f[i+1][j+1](s[j+1]=p[j+1]),f[i][j]-->f[i+1][next[j]](s[j+1]!=p[j+1]),其中next[i]是KMP中的失配数组。那么由这个转移方程我们可以O(n)的计算出答案,但是n太大了,考虑优化DP,优化DP的途径也就那么几个:斜率优化,滚动数组,矩阵乘法,更NB的数据结构。这道题我们比较容易看出是用矩阵乘法优化。因为我们设计的状态虽然是二维的,不过我们发现i的这一维只不过是限制长度的作用,并不影响转移,因此这个DP的转移矩阵是唯一的,转移矩阵a[i][j]的值即为不吉利数字串的前i位,即将匹配第i+1位,匹配的字符是0--9,所求出的next[i]=j的个数。
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define maxn 25
long long ans[maxn][maxn],g[maxn][maxn],f[maxn][maxn],h[maxn][maxn];
int next[maxn],n,m,p;
char s[maxn];
void make_next(void)
{
int k=0;
int len=strlen(s);
for (int i=1;i<len;i++)
{
while (k>0&&s[i]!=s[k]) k=next[k-1];
if (s[i]==s[k]) k++;
next[i]=k;
}
}
void qpower(void)
{
for (;n>0;n>>=1)
{
if (n&1)
{
memset(g,0,sizeof g);
for (int i=0;i<m;i++)
for (int j=0;j<m;j++)
for (int k=0;k<m;k++)
g[i][j]=(g[i][j]+(ans[i][k]*f[k][j])%p)%p;
memcpy(ans,g,sizeof g);
}
memset(h,0,sizeof h);
for (int i=0;i<m;i++)
for (int j=0;j<m;j++)
for (int k=0;k<m;k++)
h[i][j]=(h[i][j]+(f[i][k]*f[k][j])%p)%p;
memcpy(f,h,sizeof h);
}
}
int main()
{
scanf("%d%d%d",&n,&m,&p);
scanf("%s",s);
make_next();
for (int i=0;i<m;i++)
for (char j='0';j<='9';j++)
{
int k=i;
while (k>0&&s[k]!=j) k=next[k-1];
if (s[k]==j) k++;
f[i][k]++;
}
for (int i=0;i<=m;i++) ans[i][i]=1;
qpower();
long long cur=0;
for (int i=0;i<m;i++) cur=(cur+ans[0][i])%p;
printf("%lld\n",cur);
return 0;
}