题目描述
题解
转自: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]);
}
}