【GDOI2018模拟7.12】B

Description

给定一个3*3的网格图，一开始每个格子上都站着一个机器人。每一步机器人可以走到相邻格子或留在原地，同一个格子上可以有多个机器人。问走n步后，有多少种走法，满足每个格子上都有机器人。答案对10^9+7取模。

Input

一行一个整数n

Output

一行一个整数表示答案

Sample Input

1

Sample Output

229

Solution

这题的确是一道水题
你看我都在比赛时想到正解是不是水题

它有3*3的棋盘，也就是由9个点
一个很显然的想法，算出一个点在走那么多步的情况下到达另一个点的方案数
设$f[i]$为从一个点走到i走了那么多步的方案数
把和i相邻的点的方案累计起来即可
步数很多，用矩阵乘法
九个点分别计算，因为转移过程完全一样，所以只用矩阵乘法一次就行了
最后，枚举每个点走到哪个点，共$9!$种方法
时间复杂度$O(9^3*log_2(n)+9!)$

Code

#include <cstdio>
#include <algorithm>
#include <cstring>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define N 10
#define mo 1000000007
#define ll long long
using namespace std;
int dd[5][2]={0,0,1,0,-1,0,0,1,0,-1},e[N],bz[N];
ll ans,d[N][N],a[N][N],b[N][N],c[N][N],n;
int get(int x,int y){return (x-1)*3+y;}
void cl()
{
    fo(i,1,9) fo(j,1,9) c[i][j]=a[i][j],a[i][j]=0;
}
void ch1()
{
    cl();
    fo(i,1,9) fo(j,1,9) fo(k,1,9) a[i][k]=(a[i][k]+c[i][j]*b[j][k])%mo;
}
void ch()
{
    cl();
    fo(i,1,9) fo(j,1,9) fo(k,1,9) a[i][k]=(a[i][k]+c[i][j]*c[j][k])%mo;
}
void mi(ll x)
{
    if(x<2) return;
    fo(i,1,9) fo(j,1,9) b[i][j]=a[i][j];
    mi(x/2);
    ch();
    if(x%2==1) ch1();
}
void dg(int x)
{
    if(x>9)
    {
        ll an=1;
        fo(i,1,9) an=(an*d[i][e[i]])%mo;
        ans=(ans+an)%mo;
        return;
    }
    fo(i,1,9) if(!bz[i]) bz[i]=1,e[x]=i,dg(x+1),bz[i]=0;
}
int main()
{
    scanf("%lld",&n);
    fo(i,1,3) fo(j,1,3)
    {
        fo(k,0,4)
        {
            int x=i+dd[k][0],y=j+dd[k][1];
            if(x>0&&y>0&&x<4&&y<4)
            {
                a[get(i,j)][get(x,y)]=1;
            }
        }
    }
    mi(n);
    fo(i,1,9) fo(j,1,9) b[i][j]=a[i][j];
    fo(i,1,9)
    {
        memset(a,0,sizeof(a));
        a[1][i]=1;
        ch1();
        fo(j,1,9) d[i][j]=a[1][j];
    }
    ans=0;
    dg(1);
    printf("%lld\n",ans);
}