HDU 6185 Covering（dp+矩阵快速幂）

Description

用$1\times 2$和$2\times 1$的砖块铺满$4\times n$的区域，问方案数

Input

多组用例，每组用例输入一整数$n(1\le n\le 10^{18})$

Output

对于每组用例，输出方案数，结果模$10^9+7$

Sample Input

1
2

Sample Output

1
5

Solution

为叙述方便，把区域转置，变成$n\times 4$的区域

给前$i-1$层全部铺满且第$i$层为上述四种情况的方案数分别记为$A[i],B[i],C[i],D[i]$，则答案为$A[n]$

一：对于$A[i]$，考虑第$i$层四个块的组成

1.由两个$1\times 2$的组成，去掉这两块即为$A[i-1]$

2.由四个$2\times1$的组成，去掉这四块即为$A[i-2]$

3.由一个$1\times 2$和两个$2\times 1$的组成，$1\times 2$的可以在左边、中间、右边，对应的分别是$B[i-1],D[i-1],B[i-1]$

故有$A[i]=A[i-1]+A[i-2]+2B[i-1]+D[i-1]$

二：对于$B[i]$，考虑第$i$层两个块的组成

1.由一个$1\times 2$的组成，去掉这一块即为$A[i-1]$

2.由两个$2\times 1$的组成，去掉这两块即为$B[i-1]$

故有$B[i]=A[i-1]+B[i-1]$

三：对于$C[i]$，考虑第$i$层两个块的组成

只能由两个$2\times 1$的组成，去掉这两块即为$D[i-1]$

四：对于$D[i]$，考虑第$i$层两个块的组成

1.由一个$1\times 2$的组成，去掉这一块即为$A[i-1]$

2.由两个$2\times 1$的组成，去掉这两块即为$C[i-1]$

综上，有以下转移方程

$\left(\begin{array}{ccl}A[n]\\ A[n-1]\\B[n]\\C[n]\\D[n]\end{array}\right)=\left(\begin{array}{ccl}1&1&2&0&1\\1&0&0&0&0 \\1&0&1&0&0\\0&0&0&0&1\\1&0&0&1&0\end{array}\right)\left(\begin{array}{ccl}A[n-1]\\ A[n-2]\\B[n-1]\\C[n-1]\\D[n-1]\end{array}\right)=\left(\begin{array}{ccl}1&1&2&0&1\\1&0&0&0&0 \\1&0&1&0&0\\0&0&0&0&1\\1&0&0&1&0\end{array}\right)^{n-1}\left(\begin{array}{ccl}A[1]\\ A[0]\\B[1]\\C[1]\\D[1]\end{array}\right)$

其中$A[0]=A[1]=B[1]=D[1]=1,C[1]=0$

矩阵快速幂转移即可

Code

#include<cstdio>
using namespace std;
#define mod 1000000007
typedef long long ll;
typedef ll M[6][6];
M A={{1,1,2,0,1},{1,0,0,0,0},{1,0,1,0,0},{0,0,0,0,1},{1,0,0,1,0}},B;
void Mul(M &X,M &Y)
{
    M Z;
    for(int i=0;i<5;i++)
        for(int j=0;j<5;j++)
        {
            Z[i][j]=0;
            for(int k=0;k<5;k++)Z[i][j]+=X[i][k]*Y[k][j];
        }
    for(int i=0;i<5;i++)
        for(int j=0;j<5;j++)
            X[i][j]=Z[i][j]%mod;
}
void Pow(M &X,ll k)
{
    M Y;
    for(int i=0;i<5;i++)
        for(int j=0;j<5;j++)
            Y[i][j]=(i==j);
    while(k)
    {
        if(k&1)Mul(Y,X);
        Mul(X,X);
        k>>=1;
    }
    for(int i=0;i<5;i++)
        for(int j=0;j<5;j++)
            X[i][j]=Y[i][j];
}
int main()
{
    ll n;
    while(~scanf("%I64d",&n))
    {
        if(n<2)printf("1\n");
        else
        {
            for(int i=0;i<5;i++)
                for(int j=0;j<5;j++)
                    B[i][j]=A[i][j];
            Pow(B,n-1); 
            ll ans=B[0][0]+B[0][1]+B[0][2]+B[0][4];
            printf("%I64d\n",ans%mod);
        }
    }
    return 0;
}