AC日记——[HNOI2008]GT考试 bzoj 1009

1009

思路：

　　KMP上走DP(矩阵加速)；

　　DP[i][j]表示当前在第i位，同是匹配到不吉利串的第j位的方案数；

代码：

#include <bits/stdc++.h>
using namespace std;
int mod;
struct MatrixType {
    int n,m,ai[][];
    void mem(int n_,int m_)
    {
        n=n_,m=m_;
        for(int i=;i<=n;i++)
        {
            for(int v=;v<=m;v++) ai[i][v]=;
        }
    }
    MatrixType operator*(const MatrixType pos)const
    {
        MatrixType res;
        res.mem(n,pos.m);
        for(int i=;i<=res.n;i++)
        {
            for(int k=;k<=res.m;k++)
            {
                for(int v=;v<=m;v++) res.ai[i][k]=(res.ai[i][k]+(ai[i][v]*pos.ai[v][k])%mod)%mod;
            }
        }
        return res;
    }
    void debug()
    {
        puts("");
        for(int i=;i<=n;i++)
        {
            for(int v=;v<=m;v++) printf("%d ",ai[i][v]);
            putchar('\n');
        }
        puts("");
    }
};
int n,m,ti[],Next[],last[],to[][];
void poww(MatrixType res,MatrixType pos,int mi)
{
    while(mi)
    {
        if(mi&) res=res*pos;
        mi=mi>>,pos=pos*pos;
    }
    long long ans=;
    for(int i=;i<m;i++) ans=(ans+res.ai[][i])%mod;
    cout<<ans;
}
int main()
{
    freopen("data.txt","r",stdin);
    scanf("%d%d%d",&n,&m,&mod);
    char ch[];scanf("%s",ch+);
    for(int i=;i<=m;i++) ti[i]=ch[i]-'';
    for(int i=;i<m;i++)
    {
        int v=Next[i];
        while(v&&ti[i+]!=ti[v+]) v=Next[v];
        Next[i+]=ti[i+]==ti[v+]?v+:;
    }
    for(int i=;i<m;i++)
    {
        for(int v=;v<=;v++)
        {
            to[i][v]=ti[i+]==v?i+:to[Next[i]][v];
        }
    }
    MatrixType sta,tmp;
    sta.mem(,m-),sta.ai[][]=;
    tmp.mem(m-,m-);
    for(int i=;i<m;i++)
    {
        for(int v=;v<=;v++)
        {
            if(to[i][v]!=m) tmp.ai[i][to[i][v]]++;
        }
    }
    poww(sta,tmp,n);
    return ;
}