炮兵阵地(POJ 1185状压dp)

题意: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;

}

  

 

你可能感兴趣的:(poj)