Description
用 1×2 1 × 2 和 2×1 2 × 1 的砖块铺满 4×n 4 × n 的区域,问方案数
Input
多组用例,每组用例输入一整数 n(1≤n≤1018) n ( 1 ≤ n ≤ 10 18 )
Output
对于每组用例,输出方案数,结果模 109+7 10 9 + 7
Sample Input
1
2
Sample Output
1
5
Solution
为叙述方便,把区域转置,变成
n×4
n
×
4
的区域
给前 i−1 i − 1 层全部铺满且第 i i 层为上述四种情况的方案数分别记为,则答案为 A[n] A [ n ]
一:对于 A[i] A [ i ] ,考虑第 i i 层四个块的组成
1.由两个的组成,去掉这两块即为 A[i−1] A [ i − 1 ]
2.由四个 2×1 2 × 1 的组成,去掉这四块即为 A[i−2] A [ i − 2 ]
3.由一个 1×2 1 × 2 和两个 2×1 2 × 1 的组成, 1×2 1 × 2 的可以在左边、中间、右边,对应的分别是 B[i−1],D[i−1],B[i−1] B [ i − 1 ] , D [ i − 1 ] , B [ i − 1 ]
故有 A[i]=A[i−1]+A[i−2]+2B[i−1]+D[i−1] A [ i ] = A [ i − 1 ] + A [ i − 2 ] + 2 B [ i − 1 ] + D [ i − 1 ]
二:对于 B[i] B [ i ] ,考虑第 i i 层两个块的组成
1.由一个的组成,去掉这一块即为 A[i−1] A [ i − 1 ]
2.由两个 2×1 2 × 1 的组成,去掉这两块即为 B[i−1] B [ i − 1 ]
故有 B[i]=A[i−1]+B[i−1] B [ i ] = A [ i − 1 ] + B [ i − 1 ]
三:对于 C[i] C [ i ] ,考虑第 i i 层两个块的组成
只能由两个的组成,去掉这两块即为 D[i−1] D [ i − 1 ]
四:对于 D[i] D [ i ] ,考虑第 i i 层两个块的组成
1.由一个的组成,去掉这一块即为 A[i−1] A [ i − 1 ]
2.由两个 2×1 2 × 1 的组成,去掉这两块即为 C[i−1] C [ i − 1 ]
综上,有以下转移方程
⎛⎝⎜⎜⎜⎜⎜⎜A[n]A[n−1]B[n]C[n]D[n]⎞⎠⎟⎟⎟⎟⎟⎟=⎛⎝⎜⎜⎜⎜⎜⎜1110110000201000000110010⎞⎠⎟⎟⎟⎟⎟⎟⎛⎝⎜⎜⎜⎜⎜⎜A[n−1]A[n−2]B[n−1]C[n−1]D[n−1]⎞⎠⎟⎟⎟⎟⎟⎟=⎛⎝⎜⎜⎜⎜⎜⎜1110110000201000000110010⎞⎠⎟⎟⎟⎟⎟⎟n−1⎛⎝⎜⎜⎜⎜⎜⎜A[1]A[0]B[1]C[1]D[1]⎞⎠⎟⎟⎟⎟⎟⎟ ( A [ n ] A [ n − 1 ] B [ n ] C [ n ] D [ n ] ) = ( 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 ) ( A [ n − 1 ] A [ n − 2 ] B [ n − 1 ] C [ n − 1 ] D [ n − 1 ] ) = ( 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 ) n − 1 ( A [ 1 ] A [ 0 ] B [ 1 ] C [ 1 ] D [ 1 ] )
其中 A[0]=A[1]=B[1]=D[1]=1,C[1]=0 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;
}