[HDU2604]Queuing（dp+矩阵快速幂）

题目描述

传送门

题解

转自：http://blog.chinaunix.net/uid-24323834-id-261400.html
设f(n)为字符串长度为n时复合条件的字符串个数，以字符串最后一个字符为分界点，当最后一个字符为m时前n-1个字符没有限制，即为f(n-1)；当最后一个字符为f时就必须去除最后3个字符是fmf和fff的情况,在考虑最后两个字符为mf和ff的情况，显然不行；最后3个字符为fmf、mmf和fff、mff时只有当最后3个字符为mmf时前n-3个字符没有限制，即为f(n-3)，当为mff时第n-3个字符可能为f因而对前n-3个字符串有限制；最后4个字符为fmff和mmff时mmff可行。这样就讨论完了字符串的构成情况，得出结论：
f(n)=f(n-1)+f(n-3)+f(n-4)
然后就像fibonacci那样构建矩阵用快速幂取模。。。

上面讨论字符串的构成有点麻烦，可以直接用trie图搞定，由字符串fmf和fff构建trie图：

假定根节点为字符串的最后一个字符，由根结点到根节点的回路有m、fmm、ffmm，注意这是从一个字符串的后面向前走这些字符到达另一个字符串，恰好和上面的扩展递推相一致。。。

代码

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;

int n,Mod;
struct hp{int a[5][5];}unit,m,A,ans;

inline hp cheng(hp a,hp b)
{
    hp ans;
    memset(ans.a,0,sizeof(ans.a));
    for (int i=1;i<=4;++i)
        for (int j=1;j<=4;++j)
            for (int k=1;k<=4;++k)
                ans.a[i][j]=(ans.a[i][j]+a.a[i][k]*b.a[k][j]%Mod)%Mod;
    return ans;
}
inline hp matrix_fast_pow(hp a,int p)
{
    hp ans=unit;
    for (;p;p>>=1,a=cheng(a,a))
        if (p&1)
            ans=cheng(ans,a);
    return ans;
}
int main()
{
    for (int i=1;i<=4;++i) unit.a[i][i]=1;
    m.a[1][1]=m.a[3][1]=m.a[4][1]=1;
    for (int i=2;i<=4;++i) m.a[i-1][i]=1;
    A.a[1][1]=9; A.a[1][2]=6; A.a[1][3]=4; A.a[1][4]=2;
    while (~scanf("%d%d",&n,&Mod))
    {
        if (n<=4)
        {
            printf("%d\n",A.a[1][4-n+1]%Mod);
            continue;
        }
        ans=matrix_fast_pow(m,n-4);
        ans=cheng(A,ans);
        printf("%d\n",ans.a[1][1]);
    }
}