【S1矩阵乘法】多米诺

问题描述

用2x1的多米诺骨牌填满一个4xN(1 ≤ N ≤ 109) 的棋盘，总共有多少总方案。
结果可能很大，输出mod M((0 < M ≤ 105)的值。

输入格式

一行，两个整数N 和 M

输出格式

一行，一个整数表示结果。

样例输入1：

3 10000

样例输出1：

11

样例输入2：

5 10000

样例输出2：

95

---

题解

我们先来定义三种图形（每种图形的格子只画了最右边几列，其余的部分都是一列四格，竖线我省略了。。。）
这是第一种：

这是第二种：

这是第三种：

f[i][j]表示长度为 i（如果每行长度不一，为长度最长的一行）的第j 种图形刚好填满的方案数。

对于第1 种图形，我们有以下几种摆法：

由于最后一种摆法摆在上面和下面不一样，所以算两次。
因此，f[n][1]=f[n-1][3]+f[n-2][1]+f[n-1][1]+2*f[n-1][2]

对于第2 种图形，我们有以下几种摆法：

因此，f[n][2]=f[n-1][1]+f[n-1][2]

对于第3 种图形，我们有以下几种摆法：

因此，f[n][3]=f[n-2][3]+f[n-1][1]
递推式就这样出来了。
嫌n 太大？矩阵乘法来帮忙！

---

代码

#include <cstdio>
#include <algorithm>
#include <cstring>
#define ll long long
#include <iostream>
using namespace std;
typedef ll arr[15][15];
arr z,f,A,ans;
ll p;
void ch(arr x,arr y)
{
    memset(z,0,sizeof(z));
    for(ll i=1;i<=5;i++)
        for(ll j=1;j<=5;j++)
            for(ll k=1;k<=5;k++)
                z[i][j]=(z[i][j]+x[i][k]%p*y[k][j]%p)%p;
    memcpy(x,z,sizeof(z));
}
int main()
{
    ll t,n,i,k,j;
    scanf("%lld%lld",&n,&p);
    n-=2;
    for(i=1;i<=5;i++)
        ans[i][i]=1;//printf("%lld ",f[1][i]);
f[1][1]=5,f[1][2]=1,f[1][3]=1,f[1][4]=1,f[1][5]=2;
    A[1][1]=A[1][2]=A[1][3]=A[1][5]=A[2][1]=A[3][4]=A[3][1]=A[4][3]=A[5][5]=1,A[5][1]=2;
    while(n)
    {
        if(n%2)ch(ans,A);//,putchar('Q');
        n/=2,ch(A,A);
    }
    ch(f,ans);
    printf("%lld",f[1][1]);
    return 0;
}