炮兵阵地(POJ 1185状压dp)

时间:2023-10-02 09:04:08

题意:n*m地图'H'能放'p'不能放,布兵的方格上下左右不能布兵,给你地图求最大布兵数

分析:关系到前两行,所以dp[i][j][k]第i行状态为j,i-1行状态为k时的最大布兵数, 先求出所有可行的状态,统计出其布兵数。

#include <map>
#include <set>
#include <list>
#include <cmath>
#include <queue>
#include <stack>
#include <cstdio>
#include <vector>
#include <string>
#include <cctype>
#include <complex>
#include <cassert>
#include <utility>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
typedef pair<int,int> PII;
typedef long long ll;
#define lson l,m,rt<<1
#define pi acos(-1.0)
#define rson m+1,r,rt<<11
#define All 1,N,1
#define read freopen("in.txt", "r", stdin)
const ll INFll = 0x3f3f3f3f3f3f3f3fLL;
const int INF= 0x7ffffff;
const int mod = 1000000007;
int n,m,a[200],dp[110][200][200];
int num,sum[2000],cas[2000];
bool judge(int x){
return ((x<<1)&x)||((x<<2)&x)||((x>>1)&x)||((x>>2)&x);
}
int countnum(int x){
int t=0;
while(x){
if(x&1)t++;
x>>=1;
}
return t;
}
//求预处理所有可能的状态
void init(){
num=0;
for(int i=0;i<(1<<m);++i){
if(!judge(i)){
cas[num]=i;
sum[num++]=countnum(i);
}
}
}
void solve(){
init();
memset(dp,0,sizeof(dp));
for(int i=0;i<num;++i){
if(a[0]&cas[i])continue;
dp[0][i][0]=sum[i];
}
for(int i=1;i<n;++i){
//枚举当前状态
for(int j=0;j<num;++j){
if(cas[j]&a[i])continue;
//枚举符合条件的上一行状态
for(int k=0;k<num;++k){
if(cas[k]&a[i-1])continue;
if(cas[k]&cas[j])continue;
int maxv=-1;
for(int l=0;l<num;++l){
if((cas[l]&cas[k])||(cas[l]&cas[j]))continue;
maxv=max(maxv,dp[i-1][k][l]);
}
dp[i][j][k]=maxv+sum[j];
}
}
}
int maxn=-1;
for(int i=0;i<num;++i)
for(int j=0;j<num;++j){
maxn=max(maxn,dp[n-1][i][j]);
}
printf("%d\n",maxn);
}
int main()
{
char ch;
while(~scanf("%d%d",&n,&m)){
memset(a,0,sizeof(a));
for(int i=0;i<n;++i)
for(int j=0;j<m;++j){
cin>>ch;
if(ch=='H'){
a[i]|=(1<<j);
}
}
solve();
}
return 0;
}