天天看點

hdu2604(遞推,矩陣快速幂)

題目連結:hdu2604

這題重要的遞推公式,找到公式就很easy了(這道題和hdu1757(題解)類似,隻是這道題需要自己推公式)

可以直接找規律,推出遞推公式,也有另一種找遞推公式的方法:(PS:在别的部落格粘過來,暫時還不太了解。。。)

設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)

然後用矩陣快速幂就OK了~~

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <cstdlib>
using namespace std;
int n,mod;
int a[6] = {1,2,4,6,9,15};
struct node
{
    int map[4][4];
}unit,s;
void initial()//初始化
{
    int i;
    memset(s.map,0,sizeof(node));
    for(i = 1; i < 4; i ++)
    s.map[i][i-1] = 1;
    s.map[0][0] = s.map[0][2] = s.map[0][3] = 1;

    memset(unit.map,0,sizeof(node));
    for(i = 0; i < 4; i ++)//機關矩陣
    unit.map[i][i] = 1;
}
node Mul(node a,node b)
{
    node c;
    int i,j,k;
    for(i = 0; i < 4; i ++)
    for(j = 0; j < 4; j ++)
    {
        c.map[i][j] = 0;
        for(k = 0; k < 4; k ++)
        c.map[i][j] += (a.map[i][k]*b.map[k][j])%mod;
        c.map[i][j] %= mod;
    }
    return c;
}
void Matrix()
{
    while(n)
    {
        if(n&1) unit = Mul(unit,s);
        n >>= 1;
        s = Mul(s,s);
    }
    int ans = 0;
    for(int i = 0; i < 4; i ++)
    ans += (unit.map[0][i]*a[5-i])%mod;
    printf("%d\n",ans%mod);
}
int main()
{
    while(~scanf("%d%d",&n,&mod))
    {
        if(n <= 5)
        {
            printf("%d\n",a[n]%mod);
            continue;
        }
        n -= 5;
        initial();
        Matrix();
    }
    return 0;
}