刚学习了插头dp,找了道基础题练练手。
用一个n+1位的二进制数表示轮廓线上的状态,1代表有插头,0代表没有插头。
那么状态的转移为:
如果当前格的两个轮廓线上只有一个插头,那么dp[i][j][k]=dp[i][j-1][k]+dp[i][j-1][k^x^y],否则dp[i][j][k]=dp[i][j-1][k^x^y];代码如下:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int n,m;
int mp[13][13];
long long dp[13][13][5000]; //状态1表示有插头,0表示无插头
void work() {
memset(dp,0,sizeof(dp));
dp[0][m][0]=1;
int all=1<<(m+1);
for (int i=1; i<=n; i++) {
for (int k=0; k<(all>>1); k++) dp[i][0][k<<1]=dp[i-1][m][k]; //初始化dp[i][0][k]
for (int j=1; j<=m; j++) {
int x=1<<(j-1); //第j位
int y=1<<j; //第j+1位
for (int k=0; k<all; k++) {
if (mp[i][j]) {
dp[i][j][k]+=dp[i][j-1][k^x^y];
if (((k&x)>0&&(k&y)==0)||((k&x)==0&&(k&y)>0)) dp[i][j][k]+=dp[i][j-1][k]; //当前格只有一个插头
} else {
if ((x&k)||(y&k)) dp[i][j][k]=0; //当前格没有树,则不能有插头
else dp[i][j][k]=dp[i][j-1][k];
}
}
}
}
}
int main() {
int t;
scanf("%d",&t);
for (int cas=1; cas<=t; cas++) {
scanf("%d%d",&n,&m);
for (int i=1; i<=n; i++)
for (int j=1; j<=m; j++)
scanf("%d",&mp[i][j]);
work();
printf("Case %d: There are %I64d ways to eat the trees.\n",cas,dp[n][m][0]);
}
return 0;
}