1009: [HNOI2008]GT考试
Time Limit: 1 Sec Memory Limit: 162 MB
Submit: 2230 Solved: 1364
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
题解:
首先想到dp,然后看到数据范围,于是自然想到矩阵乘。
对于转移找位置,可以使用kmp加速一下(据说可以暴力,然而lcomyn、Rivendell和fye大神写的都是AC自动机。。)
不知为何矩阵乘写错了有些悲伤。。在大神指导下写了一个奇怪的递归的矩阵乘。
Code:
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 30
char s[N];
int n,m,p,l,ans=0,next[N],a[N][N],b[N][N],f[N][N];
void init(){
l=strlen(s)+1;
for (int i=l-2; i>=0; i--) s[i+1]=s[i];
s[0]=' ';
for (int i=2; i<l; i++){
int x=next[i-1];
while (x && s[i]!=s[x+1]) x=next[x];
if (s[i]==s[x+1]) next[i]=x+1;
}
for (int i=0; i<l-1; i++)
for (int j=0; j<=9; j++){
int x=i;
while (x && j!=s[x+1]-'0') x=next[x];
if (j==s[x+1]-'0') f[i][x+1]++;
else f[i][0]++;
}
memcpy(b,f,sizeof(f));
}
void jzc(int x){
if (x==1) return;
if (x>3) jzc(x>>1);
memset(a,0,sizeof(a));
for (int i=0; i<m; i++)
for (int j=0; j<m; j++)
for (int k=0; k<m; k++)
a[i][j]=(a[i][j]+f[i][k]*f[k][j])%p;
memcpy(f,a,sizeof(a));
if (x&1){
memset(a,0,sizeof(a));
for (int i=0; i<m; i++)
for (int j=0; j<m; j++)
for (int k=0; k<m; k++)
a[i][j]=(a[i][j]+f[i][k]*b[k][j])%p;
memcpy(f,a,sizeof(f));
}
}
int main(){
scanf("%d%d%d%s",&n,&m,&p,&s);
init(); jzc(n);
for (int i=0; i<m; i++)
ans=(ans+f[0][i])%p;
printf("%d\n",ans);
return 0;
}