poj1185状态压缩DP

炮兵阵地

Time Limit: 2000MS		Memory Limit: 65536K
Total Submissions: 19991		Accepted: 7737

Description

司令部的将军们打算在N*M的网格地图上部署他们的炮兵部队。一个N*M的地图由N行M列组成，地图的每一格可能是山地（用"H" 表示），也可能是平原（用"P"表示），如下图。在每一格平原地形上最多可以布置一支炮兵部队（山地上不能够部署炮兵部队）；一支炮兵部队在地图上的攻击范围如图中黑色区域所示： 

如果在地图中的灰色所标识的平原上部署一支炮兵部队，则图中的黑色的网格表示它能够攻击到的区域：沿横向左右各两格，沿纵向上下各两格。图上其它白色网格均攻击不到。从图上可见炮兵的攻击范围不受地形的影响。 
现在，将军们规划如何部署炮兵部队，在防止误伤的前提下（保证任何两支炮兵部队之间不能互相攻击，即任何一支炮兵部队都不在其他支炮兵部队的攻击范围内），在整个地图区域内最多能够摆放多少我军的炮兵部队。 

Input

第一行包含两个由空格分割开的正整数，分别表示N和M； 
接下来的N行，每一行含有连续的M个字符('P'或者'H')，中间没有空格。按顺序表示地图中每一行的数据。N <= 100；M <= 10。

Output

仅一行，包含一个整数K，表示最多能摆放的炮兵部队的数量。

Sample Input

5 4
PHPP
PPHH
PPPP
PHPP
PHHP

Sample Output

6

#include <stdio.h>
#include <string.h>
#include <string>
using namespace std;

#define max(a, b) (a) > (b) ? (a) : (b)
int m, n, top;
int state[1024], cur[110], num[1024];
int dp[150][100][100]; 
//每一行放大炮的状态，只与它上面的一行和上上一行的状态有关，每一行用状态压缩的表示方法。
//因为是和前两行的状态有关，所以要开个三维的数组表示状态，当前行的状态可由前两行的状态转移而来。即如果当前行的状态
//符合前两行的约束条件，则当前行的最大值就是上一个状态的值加上当前状态中1的个数

void init()
{
	int i;
	int total = 1 << m;
	top = 0;

	for(i = 0; i < total; i++) // for(i = 1; i <= total; i++) is wrong, find the reason
	{
		if((i&(i<<1)) || (i&i<<2))
			continue;
		state[++top]=i;
	}
}

bool isOk(int x, int id)
{	
	if(x&cur[id]) return false;
	
	return true;
}

int numOfOne(int x)
{
	int count = 0;

	while(x)
	{
		x = (x & (x-1));
		count++;
	}

	return count;
}
int main()
{
	int i, j, k, t;
	char tmp[12];

	while(scanf("%d%d", &n, &m) != EOF)
	{
		init();
		memset(dp, -1, sizeof(dp));

		for(i = 1; i <= n; i++)
		{
			memset(tmp, 0, sizeof(tmp));
			cur[i] = 0;

			scanf("%s", tmp);

			for(j = 0; tmp[j] != 0; j++)
				if(tmp[j] == 'H')
					cur[i] += (1<<(m-j-1));
		}

		//printf("%d\n", top);
		for(i = 1; i <= top; i++)
		{
			num[i] = numOfOne(state[i]);
			//printf("%d\n", num[i]);
			if(isOk(state[i], 1))
				dp[1][1][i] = num[i];
		}

		for(i = 2; i <= n; i++)
		{
			for(t = 1; t <= top; t++)
			{
				if(!isOk(state[t], i))	continue;

				for(j = 1; j <= top; j++)
				{
					if(state[t]&state[j]) continue;

					for(k = 1; k <= top; k++)
					{
						if(state[t]&state[k]) continue;
						if(dp[i-1][k][j] == -1) continue;
						dp[i][j][t]=max(dp[i][j][t], dp[i-1][k][j]+num[t]);
					}
				}
			}
		}

		int ans = 0;
		for(i = 1; i <= n; i++)
			for(j = 1; j <= top; j++)
				for(k = 1; k <= top; k++)
					ans = max(ans, dp[i][j][k]);
		printf("%d\n", ans);
	}
	return 0;
}