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

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

Time Limit: 10000/5000 MS (Java/Others)    Memory Limit: 65535/32768 K (Java/Others)

Total Submission(s): 2556    Accepted Submission(s): 902

Problem Description

　　郑厂长不是正厂长
　　也不是副厂长
　　他根本就不是厂长
　　事实上
　　他是带兵打仗的团长

　　一天，郑厂长带着他的军队来到了一个n*m的平原准备布阵。
　　根据以往的战斗经验，每个士兵可以攻击到并且只能攻击到与之曼哈顿距离为2的位置以及士兵本身所在的位置。当然，一个士兵不能站在另外一个士兵所能攻击到的位置，同时因为地形的原因平原上也不是每一个位置都可以安排士兵。
　　现在，已知n,m 以及平原阵地的具体地形，请你帮助郑厂长计算该阵地,最多能安排多少个士兵。

 

Input

输入包含多组测试数据；
每组数据的第一行包含2个整数n和m (n <= 100, m <= 10 )，之间用空格隔开；
接下来的n行，每行m个数，表示n*m的矩形阵地，其中1表示该位置可以安排士兵，0表示该地形不允许安排士兵。

 

Output

请为每组数据计算并输出最多能安排的士兵数量，每组数据输出一行。

 

Sample Input

  
  

   
   6 6
0 0 0 0 0 0
0 0 0 0 0 0
0 0 1 1 0 0
0 0 0 0 0 0
0 0 0 0 0 0
0 0 0 0 0 0
  
  

 

Sample Output

  
  

   
   2
  
  

 

题解：/(ㄒoㄒ)/~~可能的状态个数计算少了，wa了好几遍，最后调试发现m==10的时候，最大的状态数为169。

这一题和POJ 1185差不多是一样的，具体可以看POJ 1185的题解：请点击  

我们用dp[i][j][k]表示在第i行状态为k，第i-1行状态为j的情况下可以排布的士兵的最多数目。

代码如下：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int map[110][15];
int n,m,num[180];//最多的状态为169个（m==10） 
int rec[110];
int state[180],top;
int dp[110][180][180];//dp[i][j][k]表示第i行状态为k，第i-1行状态为j时的最多士兵个数 

bool ok(int x)//判断状态中有没有距离为2的1 
{
	if(x&(x<<2)) return false;
	return  true;
}

bool two(int x,int y)//判断在两个状态中有没有距离为1的1，即上下两行哈密顿路径为2 
{
	if(((x<<1)&y)||((y<<1)&x))
		return false;
	return true;
}

void init()
{
	top=0;
	for(int i=0;i<(1<<m);++i)
		if(ok(i))
			state[top++]=i; 
}

bool fit(int x,int k)//判断初始状态与当前状态是否冲突 
{
	if(rec[k]&x)	return false;
	return true;
}

int findcount(int x)//统计当前状态中1的个数 
{
	int cnt=0;
	while(x)
	{
		cnt++;
		x&=(x-1);
	}
	return cnt;
}

int main()
{
	int i,j,k,t;
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		init();
		for(i=0;i<n;++i)
		{
			rec[i]=0;
			for(j=0;j<m;++j)
			{
				scanf("%d",&map[i][j]);
				if(!map[i][j])
					rec[i]+=(1<<j);
			}
		}
		memset(dp,-1,sizeof(dp));
		for(i=0;i<top;++i)//统计每种状态中1的个数 
		{
			num[i]=findcount(state[i]);
			if(fit(state[i],0))//初始化第0行的状态 
				dp[0][0][i]=num[i];
		}
		for(i=1;i<n;++i)
		{
			for(k=0;k<top;++k)//第i行 
			{
				if(!fit(state[k],i))//判断i行状态与i行初始状态是否冲突 
					continue;
				for(j=0;j<top;++j)//第i-1行 
				{
					if(!two(state[k],state[j]))//i行和i-1行状态是否冲突 
						continue;
					for(t=0;t<top;++t)//第i-2行 
					{
						if(state[t]&state[k])//i行和i-2行不能在同一个位置上出现1 
							continue;
						if(dp[i-1][t][j]!=-1)
							dp[i][j][k]=max(dp[i][j][k],dp[i-1][t][j]+num[k]);
					}
				}
			}
		}
		int ans=0;
		for(k=0;k<top;++k)//找出所有状态中最大的个数 
		{
			for(j=0;j<top;++j)
				ans=max(ans,dp[n-1][j][k]);
		}
		printf("%d\n",ans);
	}
	return 0;
}