bzoj1009 GT考试（AC自动机+DP+矩阵快速幂）

GT考试

题目描述

阿申准备报名参加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

solution

我们可以把准考证号看做字符串，这就成了一个求不包含模式串的匹配串的个数的问题。虽然只有一个模式串，我们还是可以对它构造AC自动机，危险结点不可以经过。这样子问题就成了在一张有向图上从起点出发走n步，不经过危险结点的路径条数。观察到n非常大，而m非常小，所以可以用邻接矩阵存储trie图，然后把邻接矩阵n次方，用快速幂优化，就可以解决了

code：

#include<iostream>#include<cstdio>#include<cmath>#include<algorithm>#include<cstdlib>#include<cstring>using namespace std;int ans,n,m,k,num,x,son[1000005][10],head,tail,p,que[1000005],hz[1000005];struct matrix{    int n,a[205][205];}G;matrix operator * (matrix a,matrix b){    matrix c;    c.n=a.n;    for(int i=1;i<=a.n;i++)        for(int j=1;j<=b.n;j++){            c.a[i][j]=0;            for(int k=1;k<=b.n;k++)                c.a[i][j]=(c.a[i][j]+a.a[i][k]*b.a[k][j])%p;        }    return c;}matrix power(matrix a,int b){    matrix ans;    ans.n=a.n;    for(int i=1;i<=ans.n;i++)        ans.a[i][i]=1;    for(;b;b>>=1,a=a*a)        if(b&1)            ans=ans*a;    return ans;}char st[100];int main(){    scanf("%d%d%d",&n,&m,&p);    scanf("%s",st);    num=1;    x=1;    for(int i=0;i<m;i++)        x=son[x][st[i]-'0']=++num;    tail=0;    for(int i=0;i<=9;i++)        if(son[1][i]){            hz[son[1][i]]=1;            que[++tail]=son[1][i];        }        else            son[1][i]=1;    while(head<=tail){        int u=que[head++];        for(int i=0;i<=9;i++)            if(son[u][i]){                hz[son[u][i]]=son[hz[u]][i];                que[++tail]=son[u][i];            }            else                son[u][i]=son[hz[u]][i];    }    G.n=num-1;    for(int i=1;i<num;i++)        for(int j=0;j<=9;j++)            if(son[i][j]!=x)                G.a[i][son[i][j]]++;    G=power(G,n);    for(int i=1;i<num;i++)        if(i!=x)            ans=(ans+G.a[1][i])%p;    printf("%d\n",ans);    return 0;}