解题思路:本题是经典的状态压缩dp,由于当前行要收前两行的影响,因此我们需要一个三维的dp来解决
dp[i][t1][t2],表示当前行i的状态为t2,i-1行的状态为t1,状态转移方程dp[i][t1][t2]=max(dp[i][t1][t2],dp[i-1][j][t1]+a2[t2]) a2[t2]表示满足t2状态下可以安排的炮兵数量。(ps:做状态压缩的题目,必须先熟悉位运算的各种运算)
代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 105
char map[N][N];
int a[N],dp[N][N][N];
int a1[N],a2[N];
int slove(int x)
{
int cnt=0;
while(x)
{
if(x&1)
cnt++;
x=x>>1;
}
return cnt;
}
int main()
{
int n,m,i,j,k,t,t1,t2;
while(scanf("%d%d",&n,&m)!=EOF)
{
memset(dp,0,sizeof(dp));
for(i=0;i<n;i++)
{
scanf("%s",map[i]);
for(j=0;j<m;j++)
{
if(map[i][j]=='H')//为什么将h变为1(便于后面位运算操作),仔细想清楚
a[i]=a[i]+(1<<(m-1-j));
}
}
memset(a1,0,sizeof(a1));
memset(a2,0,sizeof(a2));
int k=0;
for(i=0;i<(1<<m);i++)
{
if((i&(i<<1))||(i&(i<<2)))continue;
a1[k]=i;
a2[k]=slove(i);
k++;
}
for(i=0;i<k;i++)
{
if(a1[i]&a[0])continue;
dp[0][0][i]=a2[i];
}
for(i=1;i<=1;i++)
{
for(j=0;j<k;j++)
{
if(a1[j]&a[i-1])continue;
for(t=0;t<k;t++)
{
if((a1[t]&a[i])||(a1[t]&a1[j]))continue;
dp[i][j][t]=max(dp[i][j][t],dp[i-1][0][j]+a2[t]);
}
}
}
for(i=2;i<n;i++)
{
for(j=0;j<k;j++)
{
if(a[i-2]&a1[j])continue;
for(t1=0;t1<k;t1++)
{
if((a[i-1]&a1[t1])||(a1[t1]&a1[j]))continue;
for(t2=0;t2<k;t2++)
{
if((a[i]&a1[t2])||(a1[t1]&a1[t2])||(a1[t2]&a1[j]))continue;
dp[i][t1][t2]=max(dp[i][t1][t2],dp[i-1][j][t1]+a2[t2]);
}
}
}
}
int ans3=0;
for(j=0;j<k;j++)
for(i=0;i<k;i++)
{
ans3=max(ans3,dp[n-1][j][i]);
}
printf("%d\n",ans3);
}
return 0;
}