poj1185 三维状压dp

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

思路：首先把所有合法状态预处理出来，可知1<<10内最多有60种合法状态，用d[i][j][k]表示第i行状态为j，第i-1行状态为k时的最大值，即遍历i-2行状态，找到i，i-1行状态确定时其最大值，状态转移方程为dp[i][j][k]=max(dp[i][j][k],dp[i-1][k][t]+sum[j])，sum[j]为j状态时1的数量

这个找bug找了一天，才发现预处理第一行时dp[1][i][0]写为dp[1][0][i]了，以后找不出bug时，可以把所有状态值都打印出来，不管状态之间转移关系，直接手算每个状态的对应值，看看是否正确。

#include <iostream>  
#include <cstdio>  
#include <cstring>  
#include <cmath>  
#include <algorithm>  
#include <stack>
#include <queue>  
#include <map>  
#include <set>  
#include <vector>  
#define LL long long  
#define eps 1e-8  
#define maxn 150  
#define mod 110119  
#define inf 0x3f3f3f3f  
#define IN freopen("in.txt","r",stdin);  
using namespace std; 

int n,m;
int MAP[105];
int cnt=0;
int str[105];
int sum[105];
int dp[105][65][65];

bool OK(int x){ //x状态是否合法，很巧妙的判断 
	if(x & x<<1 ) return false;
	if(x & x<<2 ) return false;
	return true;
}

void init(){
	int len=(1<<m);
	for(int i=0;i<len;i++){
		if(OK(i)){
			str[cnt++]=i; //把合法状态预处理出来 
		}
	}
	for(int i=0;i<cnt;i++){
		int num=0;
		int tmp=str[i];	
		while(tmp>0){
			num++;
			tmp-=tmp&-tmp;
		}
		sum[i]=num;//第i个状态对应的 1的数量 
	} 
}

int main(){
//	IN;
	cin>>n>>m;
	init();
	memset(dp,0,sizeof(dp));
	memset(MAP,0,sizeof(MAP));
	char c;
	for(int i=0;i<n;i++){
		for(int j=0;j<m;j++){
			cin>>c;
			if(c=='H'){
				MAP[i+1]|=1<<j; //也很巧妙，把不能放置的位置都设为1，取 & 为1时表示在“高地”上放置了山炮 
			}
		}
	}
	for(int i=0;i<cnt;i++){ //预处理第一行 
		if(MAP[1]&str[i])
			continue;
		dp[1][i][0]=sum[i];
	}

	for(int i=2;i<=n;i++){
		for(int j=0;j<cnt;j++){ //枚举第i行状态 
			if(MAP[i] & str[j]) continue;  
			for(int k=0;k<cnt;k++){ //枚举i-1行状态 
				if(MAP[i-1] & str[k]) continue;
				if( str[j] & str[k] ) continue;
				for(int t=0;t<cnt;t++){ //枚举i-2行状态 
					if(MAP[i-2] & str[t]) continue;
					if(str[j] & str[t] ) continue;
					if(dp[i][j][k]<dp[i-1][k][t]+sum[j]){
					} 
					dp[i][j][k]=max(dp[i][j][k],dp[i-1][k][t]+sum[j]); 
				}
			}
		}
	}
	int MAX=-1;
	for(int i=0;i<cnt;i++){
		for(int j=0;j<cnt;j++){
			MAX=max(MAX,dp[n][i][j]);
		}
	}
	cout<<MAX<<endl;
	return 0;
}