注释都在代码里。挺经典状态压缩,将二维转化成一维。结合dfs生成状态。
#include <iostream>
#include <cstdio>
using namespace std;
const long long MOD=1000000007;
long long dp[1010][64]={0};
int n,m,h;
void dfs(int base,int sta,int k,int i)
{
if(k>=m){
if(k==m){
if(base&(h>>(k)))
sta<<=1;
else
sta=(sta<<1)+1;
}
dp[i][sta]=(dp[i][sta]+dp[i-1][base>>1])%MOD;//将上一行的合法状态中的种类数加进来
return ;
}
if(base&(h>>k)) dfs(base,sta<<1,k+1,i);
else{
dfs(base,sta<<1|1,k+1,i);
if((base&(h>>k+1))==0){
if(k==(m-1)) dfs(base,sta<<2,k+2,i);
else dfs(base,sta<<2,k+2,i);
}
}
}//搜索上一行中可以产生base即j的状态,为便于处理最后一列,base=j<<1|1
int main()
{
scanf("%d%d",&n,&m);
int end=1<<m;//表示一行最多有多少种状态
h=1<<m+1;
if(m==3)
dp[1][1]=dp[1][4]=dp[1][7]=1;
else if(m==4)
dp[1][0]=dp[1][12]=dp[1][3]=dp[1][15]=dp[1][9]=1;
else{
dp[1][1]=dp[1][4]=dp[1][16]=dp[1][7]=1;
dp[1][19]=dp[1][25]=dp[1][28]=dp[1][31]=1;
}//根据m不同对第一行进行不同的初始化
for(int i=2;i<=n;i++){//从第二行开始,由上一行生成
for(int j=0;j<end;j++){
if(!dp[i-1][j]) continue;
dfs(j<<1|1,0,1,i);//j表示第i行的状态
}
}
printf("%lld\n",dp[n][0]);//对于状态的设定,可以影响到下一行同一列的设为1,否则为0
return 0;
}