BZOJ 1009 DP+矩阵乘法

        首先看到这道题，考虑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;	
}