当前位置: 代码迷 >> 综合 >> BZOJ 1009 [HNOI2008] GT考试
  详细解决方案

BZOJ 1009 [HNOI2008] GT考试

热度:57   发布时间:2024-01-19 02:22:25.0

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位的数。 N<=10^9,M<=20,K<=1000

Output

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

Sample Input

4 3 100
111

Sample Output

81

HINT

Source

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

KMP+矩阵乘法优化DP+快速幂~

(其实不用KMP也可以吧……20的数据用KMP有种鸡肋的感觉……)

先用KMP求出不吉利串的next[i],方便跳转。

转移方程比较容易想到:用f[i][j]表示目前匹配到第i位,已经有j位与不吉利串的后缀吻合的种类数,如果下一位能匹配就直接++,否则用next[i]跳转转移。

但是n比较大,用矩阵乘法+快速幂优化一下,构造b矩阵,第i位能跳转到字符j+'0'时,b[i][j]++,然后用基础矩阵a(就是对角线为1其余全为0)做快速幂就可以了,最后统计所有没有匹配出串的答案f[i][0]即可~


#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;int n,m,modd,k,next[21],ans;
char s[21];struct node{int a[21][21];
}a,b;int read()
{int totnum=0;char ch=getchar();while(ch<'0' || ch>'9') ch=getchar();while(ch>='0' && ch<='9') {totnum=(totnum<<1)+(totnum<<3)+ch-'0';ch=getchar();}return totnum;
}node operator * (node u,node v)
{node z;for(int i=0;i<m;i++)for(int j=0;j<m;j++){z.a[i][j]=0;for(int kkz=0;kkz<m;kkz++) z.a[i][j]=(z.a[i][j]+u.a[i][kkz]*v.a[kkz][j])%modd;}return z;
}int main()
{n=read();m=read();modd=read();scanf("%s",s+1);for(int i=2;i<=m;i++){while(k && s[i]!=s[k+1]) k=next[k];if(s[i]==s[k+1]) k++;next[i]=k;}for(int i=0;i<m;i++)for(int j=0;j<=9;j++){k=i;while(k && s[k+1]-'0'!=j) k=next[k];if(s[k+1]-'0'==j) k++;if(k!=m) b.a[k][i]=(b.a[k][i]+1)%modd;}for(int i=0;i<m;i++) a.a[i][i]=1;while(n){if(n&1) a=a*b;b=b*b;n>>=1;}for(int i=0;i<m;i++) ans=(ans+a.a[i][0])%modd;printf("%d\n",ans);return 0;
}