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;
}