【bzoj1009】[HNOI2008]GT考试

Description

阿申准备报名参加GT考试，准考证号为N位数X1X2….Xn(0<=Xi<=9),他不希望准考证号上出现不吉利的数字。他的不吉利数学A1A2…Am(0<=Ai<=9)有M位，不出现是指X1X2…Xn中没有恰好一段等于A1A2…Am. A1和X1可以为0

Input

第一行输入N,M,K.接下来一行输入M位的数。 100%数据N<=10^9,M<=20,K<=1000 40%数据N<=1000 10%数据N<=6

Output

阿申想知道不出现不吉利数字的号码有多少种，输出模K取余的结果.

Sample Input

4 3 100
111
Sample Output

81

代码

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
int n,m,mod;
int next[25];
char s[25];
int a[25][25],f[25][25];
using namespace std;
void mul(int a[25][25],int b[25][25],int ans[25][25])
{
    int tmp[25][25];
    for(int i=0;i<m;i++)
        for(int j=0;j<m;j++)
        {
            tmp[i][j]=0;
            for(int k=0;k<m;k++)
                tmp[i][j]=(tmp[i][j]+a[i][k]*b[k][j])%mod;
        }
    for(int i=0;i<m;i++)
        for(int j=0;j<m;j++)
            ans[i][j]=tmp[i][j];
}
int main()
{
    scanf("%d%d%d",&n,&m,&mod);
    scanf("%s",s+1);
    next[0]=-1;
    for (int i=2;i<=m;i++)
    {
        int k=i-1;
        while (s[i]!=s[next[k]+1])
        {
            k=next[k];
            if (k==0) break;
        }
        next[i]=next[k]+1;
    }
    next[0]=0;
    for (int i=0;i<m;i++)
        for (int j=0;j<=9;j++)
        {
            int t=i;
            while (t>0&&s[t+1]-'0'!=j) t=next[t];
            if (s[t+1]-'0'==j) t++;
            if (t!=m) a[t][i]=(a[t][i]+1)%mod;
        }
    for (int i=0;i<=m-1;i++)
        f[i][i]=1;
    /*for (int i=0;i<=m-1;i++)
    {
        for (int j=0;j<=m-1;j++)
            cout<<a[i][j];
         cout<<endl;
    }*/
    while(n)
    {
        if (n&1) mul(f,a,f);
        mul(a,a,a);
        n>>=1;
    }
    int sum=0;
    for (int i=0;i<m;i++)
        sum=(sum+f[i][0])%mod;
    cout<<sum;
    return 0;
}