hdu 4539 郑厂长系列故事——排兵布阵(状态压缩dp)

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

和炮兵阵地差不多

#include<stdio.h>
#include<string.h>
int tmap[105][15];
int map[105],status[200];
int dp[105][200][200];//行号、本行状态、上行状态
int n,m,up;
void getmap()
{
for(int i=2;i<n+2;i++)
{
map[i]=0;
for(int j=0;j<m;j++)
map[i]=(map[i]<<1)|(tmap[i-2][j]==0?1:0);
}
}
int getnum(int x)
{
int ans=0;
while(x)
{
if(x&1) ans++;
x>>=1;
}
return ans;
}
void getstatus()
{
up=0;
for(int i=0;i<(1<<m);i++)
{
if((i&(i<<2))) continue;
status[up++]=i;
}
}
int max(int a,int b) {return a>b?a:b;}
int main()
{
while(scanf("%d%d",&n,&m)!=EOF)
{
for(int i=0;i<n;i++)
for(int j=0;j<m;j++)
scanf("%d",&tmap[i][j]);
getmap();
getstatus();
//printf("up%d\n",up);
memset(dp,0,sizeof(dp));
int ans=0;
for(int i=2;i<n+2;i++)
for(int j=0;j<up;j++)//本行将要放的状态
if((status[j]&map[i])==0)
for(int k=0;k<up;k++)//上一行状态
if(((status[j]<<1)&status[k])==0&&(status[j]&(status[k]<<1))==0)
for(int l=0;l<up;l++)//上上行状态
if((status[j]&status[l])==0&&((status[k]<<1)&status[l])==0
&&(status[k]&(status[l]<<1))==0)
{
dp[i][j][k]=max(dp[i][j][k],dp[i-1][k][l]+getnum(status[j]));
if(dp[i][j][k]>ans)
ans=dp[i][j][k];
}

printf("%d\n",ans);

}
return 0;
}