洛谷 P3193 [HNOI2008]GT考试 题解

题目链接

分析：

下用符号$S(l,r)$表示字符串$S_l{S}_{l+1}\cdots S_r$，本文中讨论的字符串下标均从$1$开始

令$f_{i,j}$表示$X$前$i$位数字末尾$A$最多匹配到第$j$位数字（即$X(i-j+1,i)=A(1,j)$）的方案数，只要在转移时对每一个$i$都不转移$j=m$的情况（即每个前$i$位的末$m$位都不出现$A$），最终得出的答案显然就是满足题意的。

对于转移，考虑每一个$f_{i,j}$是从哪里转移来的。当前前$i$位末尾匹配到第$j$位，那么前$i-1$位末尾可以匹配到多少? 可以枚举前$i-1$位末尾能匹配到的位置$k$，则我们需要计算出$g_{k,j}$ 表示添加一位使得从最大匹配到第$k$位变成能匹配到第$j$位的方案数，即求在$S(1,k)$后添加一个数字使$S(1,j)$成为其后缀的方案，显然可以用KMP预处理出来。（其实$m⩽20$可以直接暴力求）

有$g$后转移方程就很好写了：
$$f_{i,j}=\sum _{k=0}^{m-1}(f_{i-1,k}\times g_{k,j})$$
注意到$g$是定值，显然可以用矩阵快速幂优化，令
$$F_n=\left[\begin{array}{cccc}{f}_{n,0}& f_{n,1}& \cdots & f_{n,m-1}\end{array}\right],G=\left[\begin{array}{cccc}{g}_{0,0}& g_{0,1}& \cdots & g_{0,m-1}\\ g_{1,0}& g_{1,1}& \cdots & g_{1,m-1}\\ ⋮& ⋮& \ddots & ⋮\\ g_{m-1,0}& g_{m-1,1}& \cdots & g_{m-1,m-1}\end{array}\right]$$
则有
$$F_n=F_{n-1}G$$
即
$$F_n=F_0{G}^n$$
又因为$F_0$中只有$f_{0,0}$为$1$，其它全部为$0$，最终答案只需对$G^n$的第一行求和即可

Code：

#include <iostream>
#include <cstdio>
using namespace std;
const int maxm = 25;
int n,m,ans,mod,nxt[maxm],g[maxm][maxm];
char s[maxm];
struct matrix{
    int num[maxm][maxm];
    void clear(){//全部初始化成0
        for(int i = 0; i <= m; i ++)
            for(int j = 0; j <= m; j ++)
                num[i][j] = 0;
    }
}F,G;
int read(){
    int x = 0;
    char c = getchar();
    while(c < '0' || c > '9') c = getchar();
    while(c >= '0' && c <= '9') x = x * 10 + (c ^ 48),c = getchar();
    return x;
}
matrix basic(int x){//获取x * x的单位矩阵
    matrix t;
    t.clear();
    for(int i = 0; i < m; i ++) t.num[i][i] = 1;
    return t;
}
matrix mul(matrix x,matrix y){
    matrix d;
    d.clear();
    for(int i = 0; i < m; i ++)
        for(int j = 0; j < m; j ++)
            for(int k = 0; k < m; k ++)
                d.num[i][j] = (d.num[i][j] + x.num[i][k] * y.num[k][j] % mod) % mod;
    return d;
}
matrix qpow(matrix x,long long k){
    matrix t = x,d = basic(m);
    while(k){
        if(k & 1) d = mul(d,t);
        k >>= 1,t = mul(t,t);
    }
    return d;
}
void kmp(){
    int now = 0;
     for(int i = 2; i <= m; i ++){
         while(now && s[i] != s[now + 1]) now = nxt[now];
         if(s[i] == s[now + 1]) now ++;
         nxt[i] = now;
     }
     for(int i = 0; i < m; i ++){
         for(char j = '0'; j <= '9'; j ++){
             now = i;
             while(now && s[now + 1] != j) now = nxt[now];
             if(s[now + 1] == j) now ++;
             g[i][now] ++;
         }
     }
}
int main(){
    n = read(),m = read(),mod = read();
    scanf("%s",s + 1);
    kmp();
    for(int i = 0; i < m; i ++)
        for(int j = 0; j < m; j ++)
            G.num[i][j] = g[i][j];
    G = qpow(G,n);
    for(int i = 0; i < m; i ++) ans = (ans + G.num[0][i]) % mod;
    cout << ans << endl;
    return 0;
}