【题解】HNOI2008GT考试

这题好难啊……完全不懂矩阵加速递推的我TAT

这道题目要求我们求出不含不吉利数字的字符串总数，那么我们有dp方程 ： dp[i][j](长度为 i 的字符串，最长与不吉利数字前缀相同的后缀长度为 j 的方案数)。 dp[i][j] = Σdp[i - 1][k] * a[k][j] (a 数组表示从 k 状态转移到 j 状态的方案数)。a 数组我们可以通过 kmp 对不吉利数字的每一个前缀后面加上‘0’~‘9’转移匹配得到（匹配成功表示成功转移状态，a[k][j]++;否则表示此时没有重合的后缀，a[k][0]++）。

此时这道题目我们已经拥有了一个相对优的解法了，但是还不够。注意到上面的式子，我们对于dp数组与a数组分别建立矩阵，dp矩阵是一个列矩阵，一列代表1~k的状态，a矩阵第 j 行上每个数分别表示a[j][k]。所以得到的答案dp[i][j]即为dp矩阵与a矩阵第 j 行的乘积。矩阵快速幂优化即可。

#include <bits/stdc++.h>
using namespace std;
#define maxn 100000
int n, m, Mod, k, ans;
char s[maxn], nxt[maxn];

int read()
{
    int x = , k = ;
    char c;
    c = getchar();
    while(c < '' || c > '') { if(c == '-') k = -; c = getchar(); }
    while(c >= '' && c <= '') x = x *  + c - '', c = getchar();
    return x * k;
}

struct Matrix
{
    int num[][];
    void init()
    {
        memset(num, , sizeof(num));
    }
    Matrix operator*(const Matrix &x)
    {
        Matrix tem;
        tem.init();
        for(int i = ; i < m; i ++)
            for(int j = ; j < m; j ++)
                for(int k = ; k < m; k ++)
                {
                    tem.num[i][j] += (num[i][k] * x.num[k][j]) % Mod;
                    tem.num[i][j] %= Mod;
                }
        return tem;
    }
}T, S;

void KMP()
{
    int j = ;
    for(int i = ; i <= m; i ++)
    {
        while(j && s[j + ] != s[i]) j = nxt[j];
        if(s[j + ] == s[i]) j ++;
        nxt[i] = j;
    }
    j = ;
    for(int i = ; i < m; i ++)
        for(int k = ; k <= ; k ++)
        {
            j = i;
            while(j && s[j + ] != (char) k + '') j = nxt[j];
            if(s[j + ] == (char) k + '') T.num[i][j + ] ++;
            else T.num[i][] ++;
        }
}

void Qpow()
{
    for(int i = ; i < m; i ++) S.num[i][i] = ;
    while(n)
    {
        if(n & ) S = S * T;
        T = T * T;
        n >>= ;
    }
}

int main()
{
    n = read(), m = read(), k = read();
    Mod = k;
    scanf("%s", s + );
    KMP();
    Qpow();
    for(int i = ; i < m; i ++)
        ans = (ans + S.num[][i]) % Mod;
    printf("%d\n", ans);
    return ;
}