参考自https://blog.csdn.net/weixin_43790248/article/details/84455541
刚开始完全不会做,搞了一天才勉强能按照别人代码的思路写出来。。
#include <cstdio>
#include <algorithm>
#include <cstring>
#define MOD 1000000007
#define MAX_COL 7
using namespace std;
typedef long long ll;
ll status[1<<MAX_COL][1<<MAX_COL];
ll res[1<<MAX_COL][1<<MAX_COL];
ll n,m;
// 检测S行状态的col列是否为空,并且col不越界
bool check(int S,int col) {
if(col>=0&&col<m&&(S&(1<<col))==0) return true;
return false;
}
// 在col1,col2列处置1
int Put(int S,int col1,int col2) {
S |= (1<<col1);
if(col2!=-1)
S |= (1<<col2);
return S;
}
// 2行初始状态、第一行当前状态,当前列数
void DFS(int S,int S1,int S2,int col) {
if(S1==(1<<m)-1) {
status[S][S2]++;
return;
}
// col列不为空
if(!check(S1,col)) {
DFS(S,S1,S2,col+1);
} else {
if(check(S1,col+1)&&check(S2,col)) {//
DFS(S,Put(S1,col,col+1),Put(S2,col,-1),col+2);
}
if(check(S1,col+1)&&check(S2,col+1)) {//
DFS(S,Put(S1,col,col+1),Put(S2,col+1,-1),col+2);
}
if(check(S2,col)&&check(S2,col-1)) {//
DFS(S,Put(S1,col,-1),Put(S2,col,col-1),col+1);
}
if(check(S2,col)&&check(S2,col+1)) {//
DFS(S,Put(S1,col,-1),Put(S2,col,col+1),col+1);
}
}
}
// 矩阵想乘
void Mul(ll A[1<<MAX_COL][1<<MAX_COL],ll B[1<<MAX_COL][1<<MAX_COL]) {
ll matrix[1<<MAX_COL][1<<MAX_COL] = {0};
for(int i = 0;i < (1<<MAX_COL);i++)
for(int j = 0;j < (1<<MAX_COL);j++)
for(int k = 0;k < (1<<MAX_COL);k++)
matrix[i][j] = (matrix[i][j] + A[i][k]*B[k][j]) % MOD;
for(int i = 0;i < (1<<MAX_COL);i++)
for(int j = 0;j < (1<<MAX_COL);j++)
A[i][j] = matrix[i][j];
}
int main() {
scanf("%lld%lld",&n,&m);
for(int i = 0;i < (1<<m);i++)
DFS(i,i,0,0);
for(int i = 0;i < (1<<m);i++)
for(int j = 0;j < (1<<m);j++) {
res[i][j] = (i==j)?1:0;//单位矩阵
}
while(n) {
if(n&1) {
Mul(res,status);
}
n >>= 1;
Mul(status,status);
}
printf("%lld\n",res[0][0]);
return 0;
}