四个机器人a b c d,在2 * 2的方格里,一开始四个机器人分别站在4个格子上,每一步机器人可以往临近的一个格子移动或留在原地(同一个格子可以有多个机器人停留),经过n步后有多少种不同的走法,使得每个毯子上都有1机器人停留。由于方法数量巨大,输出 Mod 10^9 + 7的结果。
Input
输入1个数N(0 <= N <= 10^9)
Output
输出走法的数量 Mod 10^9 + 7
Input示例
1
Output示例
9
题解
走法方案可以由矩阵计算,最后暴力枚举每个机器人结束为止,乘法原理统计答案。
代码
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod=1000000007;
ll ANS,a[5][5],ans[5][5];
int n;
void mul(ll a[5][5],ll b[5][5],ll c[5][5])
{
int tmp[5][5];
for (int i=1;i<=4;i++)
for (int j=1;j<=4;j++)
{
tmp[i][j]=0;
for (int k=1;k<=4;k++)
tmp[i][j]=(tmp[i][j]+a[i][k]*b[k][j]%mod)%mod;
}
for (int i=1;i<=4;i++)
for (int j=1;j<=4;j++)
c[i][j]=tmp[i][j];
}
int main()
{
scanf("%d",&n);
a[1][2]=a[1][3]=a[1][1]=1;
a[2][1]=a[2][2]=a[2][4]=1;
a[3][1]=a[3][3]=a[3][4]=1;
a[4][2]=a[4][3]=a[4][4]=1;
ans[1][1]=ans[2][2]=ans[3][3]=ans[4][4]=1;
while (n)
{
if (n&1) mul(ans,a,ans);
mul(a,a,a);
n>>=1;
}
for (int p1=1;p1<=4;p1++)
for (int p2=1;p2<=4;p2++)
for (int p3=1;p3<=4;p3++)
for (int p4=1;p4<=4;p4++)
{
if (p1==p2) continue;
if (p1==p3) continue;
if (p1==p4) continue;
if (p2==p3) continue;
if (p2==p4) continue;
if (p3==p4) continue;
(ANS+=ans[1][p1]*ans[2][p2]%mod*ans[3][p3]%mod*ans[4][p4]%mod)%=mod;
}
cout<<ANS;
return 0;
}