POJ 1185 炮兵阵地 状态压缩dp

#include<cstdio>
#include<cstring>
#include<iostream>
#definemax(a,b)(a)>(b)?(a):(b)

usingnamespacestd;

intdp[110][70][70];//状态
charmap[110][20];//存H-P图(山丘平原图)
intstk[70],cur[110];//stk[]存满足没有两个1的距离小于2的状态
//cur存读入的图用0、1表示后的状态

intnum[110];//存状态中1的个数
intm,n;//n行m列
intcnt;//cur存stk[]的个数，从1开始计。即满足1不能相邻和不能隔一个的状态数

intjcount(intx)//数一个整型数x的二进制中1的个数(用于初始化)
{
intsum=0;
while(x)
{
sum++;
x&=(x-1);
}
returnsum;
}

intisok(intx)//判断该状态是否合法
{
if(x&(x<<1))//判断是否有1相邻
return0;
if(x&(x<<2))//判断是否有1隔一个
return0;
return1;
}

voidjinit()//找到所有可能的合法状态
{
inti;
cnt=0;
inttotal=1<<m;//由于有m列，可以用二进制数0<=state<(1<<m)表示这些状态
for(i=0;i<total;i++)
{
if(isok(i))
stk[++cnt]=i;//注意cnt是从1开始的,stk[0]是没存的
}
}

intmain()
{
inti,j,k,t,ans;
while(scanf("%d%d",&n,&m)!=EOF)
{
if(n==0&&m==0)
break;
jinit();
memset(dp,0,sizeof(dp));
memset(cur,0,sizeof(cur));
getchar();
for(i=1;i<=n;i++)
{
scanf("%s",map[i]+1);
}
for(i=1;i<=n;i++)
{
for(j=1;j<=m;j++)
{
if(map[i][j]=='H')
{

cur[i]+=(1<<(j-1));
}
}
}
//初始化第一行状态
for(i=1;i<=cnt;i++)
{
num[i]=jcount(stk[i]);
if((stk[i]&cur[1])==0)
dp[1][1][i]=num[i];
}
for(i=2;i<=n;i++)//从第二行开始往后递推
{
for(j=1;j<=cnt;j++)//第i行选一个状态j
{
if(stk[j]&cur[i])
continue;
for(k=1;k<=cnt;k++)//第i-2行选一个状态k
{
if(stk[k]&stk[j])
continue;
for(t=1;t<=cnt;t++)//第i-1行选一个状态t
{
if(stk[j]&stk[t])
continue;
if(dp[i-1][k][t]==-1)
continue;
dp[i][t][j]=max(dp[i][t][j],dp[i-1][k][t]+num[j]);
}
}
}
}
ans=0;
for(i=1;i<=n;i++)
{
for(t=1;t<=cnt;t++)
{
for(j=1;j<=cnt;j++)
{
ans=max(ans,dp[i][t][j]);
}
}
}
printf("%d\n",ans);
}
return0;
}