jzoj5223 [GDOI2018模拟7.12]B 矩阵乘法

Description

---

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

Solution

---

n非常大，考虑logn做法
我们把九个机器人一一标号，然后根据四连通建邻接矩阵，矩阵乘法转移。最后暴力枚举一下可能的结果局面统计答案就可以了
为什么可以看看我的早期blog(￣▽￣)”

Code

---

#include <stdio.h>
#include <string.h>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define fill(x,t) memset(x,t,sizeof(x))

typedef long long LL;
const int MOD=1000000007;
const int N=10;

struct Matrix {
    LL a[10][10];
    Matrix operator *(Matrix B) {
        Matrix A=*this,C;
        fill(C.a,0);
        rep(i,1,9) rep(j,1,9) rep(k,1,9) {
            (C.a[i][j]+=A.a[i][k]*B.a[k][j]%MOD)%=MOD;
        }
        return C;
    }
} A;

int id[N][N],dx[4]={-1,1,0,0},dy[4]={0,0,-1,1};
int rec[N];
bool vis[N];
LL ans;

Matrix ksm(Matrix A,LL dep) {
    Matrix ret=A; dep-=1;
    while (dep) {
        if (dep&1) ret=ret*A;
        A=A*A; dep/=2;
    }
    return ret;
}

void dfs(int dep) {
    if (dep==10) {
        LL tmp=1;
        rep(i,1,9) tmp=(tmp*A.a[i][rec[i]])%MOD;
        ans=(ans+tmp)%MOD;
        return ;
    }
    rep(i,1,9) if (!vis[i]) {
        vis[i]=1; rec[dep]=i; dfs(dep+1); vis[i]=0;
    }
}

int main(void) {
    rep(i,1,3) rep(j,1,3) id[i][j]=++id[0][0];
    LL n; scanf("%lld",&n);
    rep(i,1,3) rep(j,1,3) {
        A.a[id[i][j]][id[i][j]]=1;
        rep(k,0,3) {
            int x=i+dx[k],y=j+dy[k];
            if (x>0&&x<=3&&y>0&&y<=3) {
                A.a[id[i][j]][id[x][y]]=1;
            }
        }
    }
    A=ksm(A,n);
    dfs(1); printf("%lld\n", ans);
    return 0;
}