【51Nod1122】机器人走方格 V4

四个机器人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;
}