NOI2001 炮兵阵地（状压dp）

题目描述

司令部的将军们打算在N*M的网格地图上部署他们的炮兵部队。一个N*M的地图由N行M列组成，地图的每一格可能是山地（用“H” 表示），也可能是平原（用“P”表示），如下图。在每一格平原地形上最多可以布置一支炮兵部队（山地上不能够部署炮兵部队）；一支炮兵部队在地图上的攻击范围如图中黑色区域所示：

如果在地图中的灰色所标识的平原上部署一支炮兵部队，则图中的黑色的网格表示它能够攻击到的区域：沿横向左右各两格，沿纵向上下各两格。图上其它白色网格均攻击不到。从图上可见炮兵的攻击范围不受地形的影响。 现在，将军们规划如何部署炮兵部队，在防止误伤的前提下（保证任何两支炮兵部队之间不能互相攻击，即任何一支炮兵部队都不在其他支炮兵部队的攻击范围内），在整个地图区域内最多能够摆放多少我军的炮兵部队。

输入输出格式

输入格式：

第一行包含两个由空格分割开的正整数，分别表示N和M；

接下来的N行，每一行含有连续的M个字符（‘P’或者‘H’），中间没有空格。按顺序表示地图中每一行的数据。N≤100；M≤10。

输出格式：

仅一行，包含一个整数K，表示最多能摆放的炮兵部队的数量。

输入输出样例

输入样例#1：

5 4
PHPP
PPHH
PPPP
PHPP
PHHP

输出样例#1：

6

之前没有做过状压dp，想做一道入门题，然后就找了这道题，结果后来才发现其实有点难。这道题的难点在于状态的表示、转移，另外还要注意需要预处理所有合法的状态，而不是每次都枚举所有状态，不然会T。

贴一发代码：

#include<bits/stdc++.h>
using namespace std;

int n,m,tot=0,ans=0,a[110],mask[1100],S[1100],dp[3][1100][1100];
char ch[110];

int getsum(int msk){
    int ret=0;
    while(msk)ret+=(msk&1),msk>>=1;
    return ret;
}
bool judge(int i,int msk){          //行数，状态 
    return !((msk&a[i])||((msk<<1)&msk)||((msk<<2)&msk));
}
void dfs(int d,int msk){
    if(d>=m){
        mask[++tot]=msk;
        S[msk]=getsum(msk);
        return;
    }
    dfs(d+1,msk);
    dfs(d+3,msk+(1<<d));
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%s",ch);
        for(int j=1;j<=m;j++)a[i]=(a[i]<<1)+(ch[j-1]=='H');
    }
    dfs(0,0);
    for(int i=1;i<=tot;i++)
      if(judge(1,mask[i]))
        dp[1][mask[i]][0]=S[mask[i]];
    for(int l=1;l<=tot;l++)
      for(int s=1;s<=tot;s++)
        if(judge(1,mask[l])&&judge(2,mask[s])&&!(mask[l]&mask[s]))
          dp[2][mask[s]][mask[l]]=S[mask[l]]+S[mask[s]];
    for(int i=3;i<=n;i++)
        for(int l=1;l<=tot;l++)
          if(judge(i-1,mask[l]))
            for(int s=1;s<=tot;s++)
              if(judge(i,mask[s])&&!(mask[l]&mask[s]))
                for(int k=1;k<=tot;k++)
                  if(judge(i-2,mask[k])&&!(mask[k]&mask[s])&&!(mask[k]&mask[l]))
                    dp[i%3][mask[s]][mask[l]]=max(dp[i%3][mask[s]][mask[l]],dp[(i-1)%3][mask[l]][mask[k]]+S[mask[s]]);
    for(int s=1;s<=tot;s++)
      for(int i=1;i<=tot;i++)
        ans=max(ans,dp[n%3][mask[s]][mask[i]]);
    printf("%d",ans);
    return 0;
}