BZOJ1009 [HNOI2008]GT考试

题意：给定长度为m(m <= 20)的数字串，求长度为n(n <= 10^9)的数字串中不包含这个数字串的个数，对k取模。

分析：

这不就是个dp吗...

n <= 10^9，显然这就是矩阵快速幂啊。

但是矩阵里存什么信息呢？

设f(i, j)表示填了i个字符，结尾最多匹配到j个字符的方案数。

f(i, j) = ∑f(i-1, k)*a(k, j)     (0 <= j,k < m)，其中a(i, j)表示匹配了前i个字符，填一个数字匹配到第j个字符的方案数。

观察到这个式子满足矩阵快速幂的形式，所以我们的重点就是求a数组。

字符串匹配，可以用KMP求这个数组，先处理出nxt数组，然后随便乱搞就行了（这里矩阵行列存反了，凑合看吧）

#include <cstdio>
#include <cstring>

char s[25];
int n,m,p,j,ans,nxt[25];

struct mat {
	int a[25][25];
	mat operator * (const mat &b) const {
		mat c;
		memset(c.a, 0, sizeof c.a);
		for(int k = 0; k < m; k++)
		for(int i = 0; i < m; i++)
		for(int j = 0; j < m; j++)
			c.a[i][j] = (c.a[i][j]+a[i][k]*b.a[k][j])%p;
		return c;
	}
}e,b;

void pw(int n) {
	while(n) {
		if(n & 1) e = e*b;
		b = b*b, n >>= 1;
	}
}

int main() {
	scanf("%d%d%d%s", &n, &m, &p, s+1);
	for(int i = 0; i < m; i++) e.a[i][i] = 1;
	for(int i = 2; i <= m; i++) {
		while(j && s[j+1] != s[i]) j = nxt[j];
		if(s[j+1] == s[i]) j++;
		nxt[i] = j;
	}
	for(int i = 0; i < m; i++)
	for(int j = 0; j <= 9; j++) {
		int h = i;
		while(h && s[h+1]-'0' != j) h = nxt[h];
		if(s[h+1]-'0' == j) h++;
		b.a[h][i] = (b.a[h][i]+1)%p;
	}
	pw(n);
	for(int i = 0; i < m; i++) ans = (ans+e.a[i][0])%p;
	printf("%d", ans);
	return 0;
}