POJ 1185 状态压缩DP 炮兵阵地

题目直达车:   POJ 1185 炮兵阵地

分析:

列( <=10 )的数据比较小, 一般会想到状压DP.

Ⅰ、如果一行10全个‘P’，满足题意的状态不超过60种（可手动枚举）。

Ⅱ、用DFS搜出所有可能表示状态的整数（二进制1表示可以放，0则不能）。

Ⅲ、对每一行的地行进行状态处理（p[i]表示第i行地形的状态），二进制‘H’转1，‘P’转0；

Ⅳ、用dp[i][j][k]表示第i行，且i行状态为j，i-1行状态为k时，最多能放置的量。

Ⅴ、对于第i行的可行状态必须满足：

⒈  j & k =0 且 j & t =0 （t为第i-2行放置状态）与前两行匹配，即不互相攻击。

⒉  j & p[i] = 0 与地形匹配，即只放平原地带。

Ⅵ、转移方程：dp[i][j][k]=max(dp[i][j][k],dp[i-1][k][t]+num[j])

（num[j]为第j种状态中放置炮兵的数量）

第一次做状态DP,  参考(题解报告)的思路才敲出来

代码:

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<vector>
using namespace std;
int p[105],dp[105][65][65];
vector<int>M[11];
int a[11],n,m;
void DFS(int k,int t){  ///搜索m=t时如果全为'P'可以表示状态的整数
    if(k==t){
        int p=0;
        for(int i=0;i<t;++i)
            if(a[i]) p=p|(1<<i);
        M[t].push_back(p);  /// 记下能表示状态的整数
        return;
    }
    for(int i=0;i<=1;++i){
        if(i&&((k&&a[k-1])||(k>1&&(a[k-1]||a[k-2])))) continue;
        a[k]=i;
        DFS(k+1,t);
    }
}
int main(){
    for(int i=1;i<=10;++i) DFS(0,i);
    while(~scanf("%d%d",&n,&m)&&n+m){
        p[0]=0;
        ///num[i]记下用第i种状态(整数M[m][i]表示) 中设了多少炮兵部队
        int num[65], len=M[m].size();
        for(int i=0;i<len;++i)
            num[i]=__builtin_popcount( M[m][i] ); ///计算M[m][i]中二进制中1的个数
        memset(dp,0,sizeof(dp));
        int Max=0;
        for(int i=1;i<=n;++i){
            char ch[105]; scanf("%s",ch);
            p[i]=0;
            //预处理地形(1代表'H',0代表'P')
            for(int j=0;j<m;++j) if(ch[j]=='H') p[i]|=(1<<j);
            if(i==1){ //初始化第一行所有布置情况
                for(int k=0; k<len; ++k)
                    for(int j=0; j<len; ++j){
                        if(!(M[m][k]&p[1])) dp[1][k][j]=num[k];
                        else dp[1][k][j]=0;
                        Max=max(Max,dp[1][k][j]);
                    }
                continue;
            }
            for(int j=0; j<len; ++j)//当前行
                if(!(M[m][j]&p[i])) //与 地形匹配
                    for(int k=0; k<len; ++k)//i-1
                        if(!(M[m][k]&p[i-1])&&!(M[m][k]&M[m][j]))//与 地形、i-1行匹配
                            for(int t=0; t<len; ++t)//i-2
                                //与 地形、i-1行、i-2行匹配
                                if(!(M[m][t]&p[i-2])&&!(M[m][t]&M[m][k])&&!(M[m][t]&M[m][j])){
                                    dp[i][j][k]=max(dp[i][j][k],dp[i-1][k][t]+num[j]);
                                    if(i==n)Max=max(dp[i][j][k],Max);
                                }
        }
        printf("%d\n",Max);
    }
    return 0;
}