POJ 1185 炮兵阵地（状态压缩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
Solution
状态压缩DP，首先显然是通过一个二进制数来表示一行的状态，然后通过前两行来推出下一行的状态，但考虑到数据范围，如果每次都暴力枚举一行所有状态会超时，所以在状态数上需要优化，因为每行最多10格，相邻两支炮兵部队最小间距为2，所以合法状态不会超过60种，这样的话我们可以预处理每行都为空地时的所有合法状态，每行枚举状态时就极大的降低了时间复杂度。
下面看具体操作，用dp[h][i][j]表示第h行状态为i，第h-1行状态为j时的炮兵总数，首先算出前两行的所有合法状态，之后由递推式dp[h][i][j]=max{dp[h][i][j],dp[h-1][j][k]+num[i]}(k为第h-2行的状态,num[i]表示第h行状态为i时可以放置的炮兵数)就可以递推出所有行的合法状态，这样最多能放置的数量即为max{dp[n-1][i][j]}(0<=i< p,0<=j< p,p为所有合法状态)
Code

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
#define maxn 111
int map[maxn];
int dp[maxn][maxn][maxn];
int s[maxn],num[maxn];
int n,m,p;
bool check(int x)//判断状态x是否会造成行冲突 
{
    if(x&(x>>1))return false; 
    if(x&(x>>2))return false;
    return true;
}
int count(int x)//统计x状态需要放置多少炮兵 
{
    int i=1,ans=0;
    while(i<=x)
    {
        if(x&i)//该位置放置炮兵 
            ans++;
        i<<=1;
    }
    return ans;
}
void init()
{
    //初始化 
    p=0;
    memset(s,0,sizeof(s));//s数组存所有合法状态 
    memset(num,0,sizeof(num));//num数组存所有合法状态的炮兵数 
    for(int i=0;i<(1<<m);i++)//枚举所有可能状态 
    {
        if(check(i))//i状态合法 
        {
            s[p]=i;//记录 
            num[p++]=count(i);//统计i状态需要放置炮兵数 
        }
    }
}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        if(!n&&!m)//有一边为0结果显然为0 
        {
            printf("0\n");
            continue;
        }
        //初始化 
        memset(dp,0,sizeof(dp));
        memset(map,0,sizeof(map));
        for(int i=0;i<n;i++)//输入 
            for(int j=0;j<m;j++)
            {
                char ch;
                cin>>ch;
                if(ch=='H')
                    map[i]=map[i]|(1<<(m-1-j));//只有P才能放炮兵 
            }
        init();//预处理 
        for(int i=0;i<p;i++)//计算第一行的状态 
        {
            if(!(map[0]&s[i]))
                dp[0][i][0]=num[i];
        }
        for(int i=0;i<p;i++)//计算第二行的状态 
        {
            if(!(map[1]&s[i]))
            {
                for(int j=0;j<p;j++)
                {
                    if(!(s[i]&s[j]))
                        dp[1][i][j]=max(dp[1][i][j],dp[0][j][0]+num[i]);
                }
            }
        }
        for(int h=2;h<n;h++)//计算第2~n-1行的状态 
        {
            for(int i=0;i<p;i++)//枚举第h行状态 
            {
                if(!(map[h]&s[i]))//由地形排除不合法状态 
                {
                    for(int j=0;j<p;j++)//枚举第h-1行状态 
                    {
                        if(!(map[h-1]&s[j]))//由地形排除不合法状态 
                        {
                            if(!(s[i]&s[j]))//第h行与第h-1行不能冲突 
                            {
                                for(int k=0;k<p;k++)//枚举第h-2行状态 
                                {
                                    if(!(map[h-2]&s[k]))//由地形排除不合法状态 
                                    {
                                        if(!(s[i]&s[k]))//第h行不能与第h-2行冲突 
                                        {
                                            if(!(s[j]&s[k]))//第h-1行不能与第h-2行冲突 
                                            {
                                                dp[h][i][j]=max(dp[h][i][j],dp[h-1][j][k]+num[i]);
                                            }
                                        }
                                    }
                                }
                            }
                        }
                    }
                }
            }
        }
        int ans=0;
        for(int i=0;i<p;i++)
            for(int j=0;j<p;j++)
                if(ans<dp[n-1][i][j])
                    ans=dp[n-1][i][j];
        printf("%d\n",ans);
    }
    return 0;
}