【洛谷 2021.11.1团队模拟赛】T202676 网格游走

---

解题思路

考虑用矩阵乘法预处理出从起点i到终点j之间的方案数。

把3*3网格从1~9标号每个点和他上下左右的点以及自己之间能互相到达

构造 9 * 9 矩阵B：（对应上面那个图直观感受）

$\left[\begin{array}{ccccccccccc}& 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{array}\right]$

初矩阵中每个点有一个机器人，所以构造j矩阵A：
$\left[\begin{array}{ccccccccccc}& 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{array}\right]$

$C=B^n\ast A$，C为最终表示起点i到终点j之间的方案数的矩阵。

然后我们$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);
}