HDU-4539-郑厂长系列故事——排兵布阵

这个题是腾讯马拉松的一个题，当时没做出来，今天看了下状态压缩，然后做了下。感觉跟POJ1185很像，基本上只是改一下。

其实主要修改的只有2个地方，因为这个地所要求的是曼哈顿距离为2，所以在转移上一层状态的时候应该判断是否左右相邻。同时上层与上上层之间也同样需要做这样一个判断。

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int maxn=202;
int dp[maxn][maxn][maxn],e[maxn],s[maxn],c[maxn];
int n,m,cnt;
int Count(int x)
{
int ans=0;
while(x)
{
if(x&1)
ans++;
x>>=1;
}
return ans;
}
int main()
{
while(scanf("%d%d",&n,&m)!=EOF)
{
memset(e,0,sizeof(e));
for(int i=0;i<n;i++)
for(int j=0;j<m;j++)
{
int ita;
scanf("%d",&ita);
if(!ita)
e[i]|=(1<<j);
}
cnt=0;
int total=1<<m;
for(int i=0;i<total;i++)
{
if(i&(i<<2))
continue;
s[cnt]=i;
c[cnt++]=Count(i);
}
memset(dp,-1,sizeof(dp));
for(int i=0;i<cnt;i++)
{
if(s[i]&e[0])
continue;
dp[0][i][0]=c[i];
}
for(int i=1;i<n;i++)
{
for(int j=0;j<cnt;j++)
{
if(e[i]&s[j])
continue;
for(int k=0;k<cnt;k++)
{
if((s[j]&(s[k]<<1))||(s[j]&(s[k]>>1)))
continue;
for(int l=0;l<cnt;l++)
{
if(s[j]&s[l])
continue;
if(s[k]&(s[l]>>1))
continue;
if(s[k]&(s[l]<<1))
continue;
if(dp[i-1][k][l]==-1)
continue;
dp[i][j][k]=max(dp[i][j][k],dp[i-1][k][l]+c[j]);
}
}
}
}
int ans=0;
for(int i=0;i<cnt;i++)
for(int j=0;j<cnt;j++)
ans=max(ans,dp[n-1][i][j]);
printf("%d\n",ans);
}
return 0;
}