[bzoj 1009] [HNOI2008]GT考试：DP，单串AC自动机，矩阵快速幂

题意：给一个长为M（M<=20）的十进制数字串，求有多少个长为N（N<=10^9）的十进制数字串不包含它，结果对K（K<=10^3）取模。

在黄学长博客的KMP分类中找到这题。并没产生什么好思路……准确地说是没深入思考，应该批评。如果把f[i][S]——前i个字符，以S为(M-1)后缀——作为状态，那么来个20维的DP是能想到的……枚举第(i+1)个字符，只要不形成给定的模板串就可以转移。

把这个暴力放到AC自动机上。这里只有一串。f[i][j]表示文本串的前i个字符匹配到Trie图前j个结点的方案数。如果有一条边(j, k)，那么f[i+1][k] += f[i][j]。矩阵快速幂优化。

另一种理解方式。每一个合法文本串对应Trie图上从起点开始的一条路径。求出Trie图的邻接矩阵。对邻接矩阵做矩阵乘法，即得任意两点间恰好经过N条边的路径数。

解法来源于[bzoj 1030] [JSOI2007]文本生成器。

用矩阵乘法优化DP的关键在于构造转移矩阵A。如果想让转移矩阵乘到右边，那么A[i][j]是i对j的贡献。如果想让转移矩阵乘到左边，那么A[i][j]是i从j得到的贡献。

我的矩阵快速幂写的不是很优雅……几个数组复制来复制去的……黄学长用迭代写的，很精简。

一开始我真的是用KMP的失配指针转移的……没仔细想。失配不一定只跳一次嘛，这样怎么能让长度为N的字符串和经过N条边的路径对应呢？其实AC自动机也有多种写法，这里要用补完版本。

放两份代码。第二份是纯暴力，可以用于检验小数据。

#include <cstdio>#include <cstring>using namespace std;int N, M, K;const int MAX_M = 20, sz = sizeof(int)*MAX_M*MAX_M, SS = 10;char a[MAX_M+1];int A[MAX_M][MAX_M], next[MAX_M][10], fail[MAX_M];void build(){    A[0][0] = SS-1;    if (M == 1)        return;    A[0][1] = 1;    next[0][a[0]] = 1;    for (int i = 1; i < M; ++i) {        if (i > 1)            fail[i] = next[fail[i-1]][a[i-1]];        for (int j = 0; j < SS; ++j) {            if (i == M-1 && j == a[i])                continue;            next[i][j] = j == a[i] ? i+1 : next[fail[i]][j];            if (++A[i][next[i][j]] == K)                A[i][next[i][j]] = 0;        }    }}void matrix_mul(int C[MAX_M][MAX_M], int A[MAX_M][MAX_M], int B[MAX_M][MAX_M]){    memset(C, 0, sz);    for (int i = 0; i < M; ++i)        for (int j = 0; j < M; ++j)            for (int k = 0; k < M; ++k)                (C[i][j] += A[i][k] * B[k][j]) %= K;    }void matrix_exp(int A[MAX_M][MAX_M], int n){    if (n == 0) {        memset(A, 0, sz);        for (int i = 0; i < M; ++i)            A[i][i] = 1;        return;    }    if (n == 1)        return;    int B[MAX_M][MAX_M], C[MAX_M][MAX_M];    memcpy(B, A, sz); // B = A    matrix_exp(A, n/2);    matrix_mul(C, A, A);    if (n & 1)        matrix_mul(A, C, B);    else        memcpy(A, C, sz); // A = C}int main(){    scanf("%d %d %d %s", &N, &M, &K, a);    for (int i = 0; i < M; ++i)        a[i] -= '0';    build();    matrix_exp(A, N);    int ans = 0;    for (int j = 0; j < M; ++j)        ans = (ans + A[0][j]) % K;    printf("%d\n", ans);    return 0;}

#include <cstdio>#include <cstring>using namespace std;const int MAX_N = 10, MAX_M = 10, SS = 10;int N, M, K;char a[MAX_M], b[MAX_N];int search(int k){    int ans = 0;    if (k == N) {        bool ok = false;        for (int i = 0; i < N; ++i)            if (i+M <= N && !strncmp(a, b+i, M)) {                ok = true;                break;            }        return !ok;    }    for (int i = 0; i < SS; ++i) {        b[k] = i+'0';        (ans += search(k+1)) %= K;    }    return ans;}int main(){    scanf("%d %d %d %s", &N, &M, &K, a);    printf("%d\n", search(0));    return 0;}