BZOJ1009: [HNOI2008]GT考试

Portal

裹了大半天，终于把这个（chu）（sheng）题搞清白了
$\require{enclose}\enclose{horizontalstrike}{（哪个准考证号10^{9}位，还挑号码，疯了板）}$

令$f[i][j]$表示 准考证号前$i$位，匹配到不吉利数的第$j$位（即当前准考证号后$j$位与不吉利数字前$j$位相同）的方案数
$ans=\sum_{i=0}^{m-1}{f[n][i]}$

令$b[i][j]$表示从不吉利串第$i$个字符后加一个字符可以从$i$个匹配到$j$个的可以填写的数字的个数。

就可得：$f[i][j]=\sum_{k=0}^{m-1}{f[i-1][k]*b[k][j]}$

发现$f[i][j]$只会由$f[i-1]$得到，$b$数组只与不吉利串本身有关系，那么$b$数组就是个常量已知矩阵。
KMP预处理出$b$数组（这里面细节想了好久，可能kmp学的不好吧hh），
其实打完以后发现..还是很好理解的。

for(int i=0,sum,j;i<m;i++)
    {
        j=i;
        sum=b.v[i][j+1]=1;
        mp[ch[j+1]-'0']=i;//直接跳到下一位有一种填数方式，标记这个数
        while(j)
        {
            j=Next[j];
            if(mp[ch[j+1]-'0']!=i)//这个数已经填过的就不管了，填了这个数可以跳到更后面的位置，所以当前是不取的
            {
                mp[ch[j+1]-'0']=i;
                b.v[i][j+1]=1;
                sum++;
            }
        }
        b.v[i][0]=10-sum;//剩下的数全部都是重新开始匹配
    }

$n$达到$10^{9}$，显然不能直接推，然后发现这是常系数线性齐次递推式。初始令$f[0][0]=1$。$|1*n|$乘$|n*n|$，就可以开开心心的矩乘啦！

【代码】

#include <iostream>
#include <cstdio>
#include <algorithm>
#define N 10005
using namespace std;
typedef long long ll;

int read()
{
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
}

int n,m,mod,ans;
int Next[25],mp[10];
char ch[25];

class matrix {
    public:
        int x,y;
        int v[22][22];
}a,b; 

matrix Matrix_Mul(matrix aa,matrix bb)
{
    matrix rtn;
    rtn.x=aa.x;rtn.y=bb.y;
    for(int i=0;i<aa.x;i++)
    for(int j=0;j<aa.y;j++)
    {
        rtn.v[i][j]=0;
        for(int k=0;k<bb.y;k++)
            rtn.v[i][j]=(rtn.v[i][j]+aa.v[i][k]*bb.v[k][j])%mod;
    }
    return rtn;
}

void Matrix_Pow(int y)
{
    while(y)
    {
        if(y&1) a=Matrix_Mul(a,b);
        b=Matrix_Mul(b,b);y>>=1;
    }
}

void KMP()
{
    Next[1]=1;
    for(int i=2,j=1;i<=m;i++)
    {
        while(j>1&&ch[j]!=ch[i]) 
            j=Next[j-1];
        if(ch[j]==ch[i]) j++;
        Next[i]=j;
    }
    for(int i=1;i<=m;i++) Next[i]--;
    for(int i=0,sum,j;i<m;i++)
    {
        j=i;
        sum=b.v[i][j+1]=1;
        mp[ch[j+1]-'0']=i;
        while(j)
        {
            j=Next[j];
            if(mp[ch[j+1]-'0']!=i)
            {
                mp[ch[j+1]-'0']=i;
                b.v[i][j+1]=1;
                sum++;
            }
        }
        b.v[i][0]=10-sum;
    }
}

void Solve()
{
    a.v[0][0]=1;
    Matrix_Pow(n);
    for(int i=0;i<m;i++) ans+=a.v[0][i];
    printf("%d\n",ans%mod);
}

int main()
{
    n=read(),m=read(),mod=read();
    scanf("%s",ch+1);a.x=1;a.y=b.x=b.y=m;
    KMP();
    Solve();
    return 0;
}