解题思路
考虑用矩阵乘法预处理出从起点i到终点j之间的方案数。
把3*3网格从1~9标号每个点和他上下左右的点以及自己之间能互相到达
构造 9 * 9 矩阵B:(对应上面那个图直观感受)
[ 0 1 0 1 0 0 0 0 0 1 0 1 0 1 0 0 0 0 0 1 0 0 0 1 0 0 0 1 0 0 0 1 0 1 0 0 0 1 0 1 0 1 0 1 0 0 0 1 0 1 0 0 0 1 0 0 0 1 0 0 0 1 0 0 0 0 0 1 0 1 0 9 0 0 0 0 0 1 0 1 0 ] \begin{bmatrix} &0 &1&0&1&0&0&0&0&0 &\\ &1&0&1&0&1&0&0&0&0 \\ &0&1&0&0&0&1&0&0&0 \\ &1&0&0&0&1&0&1&0&0 \\ &0 &1&0&1&0&1&0&1&0 \\ &0 &0&1&0&1&0&0&0&1 \\ &0 &0&0&1&0&0&0&1&0 \\ &0 &0&0&0&1&0&1&0&9 \\ &0 &0&0&0&0&1&0&1&0 \end{bmatrix} ⎣⎢⎢⎢⎢⎢⎢⎢⎢⎢⎢⎢⎢⎡010100000101010000010001000100010100010101010001010001000100010000010101000001090⎦⎥⎥⎥⎥⎥⎥⎥⎥⎥⎥⎥⎥⎤
初矩阵中每个点有一个机器人,所以构造j矩阵A:
[
1
0
0
0
0
0
0
0
0
0
1
0
0
0
0
0
0
0
0
0
1
0
0
0
0
0
0
0
0
0
1
0
0
0
0
0
0
0
0
0
1
0
0
0
0
0
0
0
0
0
1
0
0
0
0
0
0
0
0
0
1
0
0
0
0
0
0
0
0
0
1
0
0
0
0
0
0
0
0
0
1
]
\begin{bmatrix} &1 &0&0&0&0&0&0&0&0 & \\ &0 &1&0&0&0&0&0&0&0 & \\ &0 &0&1&0&0&0&0&0&0 & \\ &0 &0&0&1&0&0&0&0&0 & \\ &0 &0&0&0&1&0&0&0&0 & \\ &0 &0&0&0&0&1&0&0&0 & \\ &0 &0&0&0&0&0&1&0&0 & \\ &0 &0&0&0&0&0&0&1&0 & \\ &0 &0&0&0&0&0&0&0&1 & \end{bmatrix}
⎣⎢⎢⎢⎢⎢⎢⎢⎢⎢⎢⎢⎢⎡100000000010000000001000000000100000000010000000001000000000100000000010000000001⎦⎥⎥⎥⎥⎥⎥⎥⎥⎥⎥⎥⎥⎤
C = B n ∗ A C=B^n*A C=Bn∗A,C为最终表示起点i到终点j之间的方案数的矩阵。
然后我们 d f s dfs dfs暴力一下每个点最终到哪里,即9的全排列。然后如果对应的起点到终点都是能到达的,就统计方案。
代码
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const ll mod=1e9+7;
const ll dx[6]= {0,1,-1,0,0},dy[6]= {0,0,0,1,-1};
ll n,k,w[10][10],v[10],vis[10];
ll ans;
struct c{
int n,m;
ll a[11][11];
}A,B,C;
c operator *(c A,c B){
c CC;
CC.n=A.n,CC.m=B.m;
for(int i=1;i<=CC.n;i++)
{
for(int j=1;j<=CC.m;j++)
{
CC.a[i][j]=0;
}
}
for(int k=1;k<=A.m;k++)
{
for(int i=1;i<=A.n;i++)
{
for(int j=1;j<=B.m;j++)
{
CC.a[i][j]=(CC.a[i][j]+(A.a[i][k]*B.a[k][j])%mod)%mod;
}
}
}
return CC;
}
void poww(ll x){
if(x==1)
{
C=B;
return;
}
poww(x>>1);
C=C*C;
if(x&1)
C=C*B;
return ;
}
ll check(){
ll s=1;
for(int i=1;i<=9;i++)
{
if(!A.a[i][v[i]])
return 0;
else s=s*A.a[i][v[i]]%mod;
}
return s;
}
void dfs(int dep){
if(dep==0)
{
ans=(ans+check())%mod;
return;
}
for(int i=1;i<=9;i++)
{
if(!vis[i])
{
vis[i]=1;
v[dep]=i;
dfs(dep-1);
vis[i]=0;
}
}
}
int main() {
scanf("%lld",&n);
A.n=9,A.m=9;
for(int i=1; i<=9; i++)
A.a[i][i]=1;
B.n=9,B.m=9;
for(int i=1; i<=3; i++) {
for(int j=1; j<=3; j++)
w[i][j]=++k;
}
for(int i=1; i<=3; i++) {
for(int j=1; j<=3; j++) {
for(int u=0; u<=4; u++) {
int xx=i+dx[u],yy=j+dy[u];
if(xx<=0||yy<=0||xx>3||yy>3)continue;
int s1=w[i][j],s2=w[xx][yy];
B.a[s1][s2]=1;
}
}
}
poww(n);
A=A*C;
dfs(9);
printf("%lld",ans);
}