POJ1185——炮兵阵地（状态压缩DP）

炮兵阵地

Time Limit: 2000MS Memory Limit: 65536K

Total Submissions: 14787 Accepted: 5523

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
Source

Noi 01

解析：

        回归状压DP的第一题。。。

        dp[i][j][k] 表示第i行状态为j，第i-1状态为k时的最大炮兵个数

        dp[i][k][t] =max(dp[i][k][t],dp[i-1][j][k]+num[t]); num[t]为t状态中1的个数。。。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

int n,m,state[110];
char map[110][20];
int dp[110][70][70];
int pos[110],yi[110];

void read()
{
    freopen("poj1185.in","r",stdin);
    freopen("poj1185.out","w",stdout);
}

bool ok(int x) //检查状态x是否可行
{
    if(x&(x<<1))return 0;
    if(x&(x<<2))return 0;
    return 1;
}

bool check(int i,int s) //检查第i行和状态s是否可行
{
    if(state[i]&s)return 0;
    return 1;
}

int count(int x)//计算状态x中1的个数
{
    int cnt=0;
    while(x)
    {
        if(x&1)cnt++;
        x>>=1;
    }
    return cnt;
}

void work()
{
    scanf("%d%d",&n,&m);
    int tot=1<<m;  int num=0;
    for(int i=0;i<tot;i++)if(ok(i))pos[++num]=i;//找出所有可行状态，放在pos中
    for(int i=1;i<=n;i++)scanf("%s",map[i]+1);
    for(int i=1;i<=n;i++)
    {
        state[i]=0;
        for(int j=1;j<=m;j++)
            if(map[i][j]=='H')state[i]+=(1<<(j-1));
    }
    memset(dp,-1,sizeof(dp));
    for(int j=1;j<=num;j++)
    {
        yi[j]=count(pos[j]);
        if(check(1,pos[j]))
            dp[1][j][1]=yi[j];  //边界
    }
    for(int i=2;i<=n;i++)
    {
        for(int s1=1;s1<=num;s1++)
        {
            if(!check(i,pos[s1]))continue;            //  行     状态
            for(int s3=1;s3<=num;s3++)           //  i-2     s3
            {                                    //  i-1     s2
                if(pos[s1]&pos[s3])continue;     //  i       s1
                for(int s2=1;s2<=num;s2++)
                {
                    if(pos[s1]&pos[s2])continue;
                    if(dp[i-1][s2][s3]==-1)continue;
                    dp[i][s1][s2]=max(dp[i][s1][s2],dp[i-1][s2][s3]+yi[s1]);
                }
            }
        }
    }
    int ans=0;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=num;j++)
            for(int k=1;k<=num;k++)
            ans=max(ans,dp[i][j][k]);
    printf("%d\n",ans);
}

int main()
{
    read();
    work();
    return 0;
}