【POJ1185】炮兵阵地（状压DP）

题意：

思路：状压DP经典题

可以预处理下每一行内合法的状态，发现很少

所以转移时可以使用状态的编号而不是状态本身

DP时记录前两行状态的编号进行转移和判断

#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<iostream>
#include<cstring>
#include<map>
#include<set>
#include<vector>
#define INF 10000
#define MAXK 65
using namespace std;
int dp[110][MAXK][MAXK];
int a[MAXK],num[MAXK];
int b[110];
int n,m,K; 
char s[12];

int main()
{
    freopen("poj1185.in","r",stdin);
    freopen("poj1185.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",s+1);
        for(int j=1;j<=m;j++) 
         if(s[j]=='H') b[i]+=(1<<(j-1));
    }
    int MAX=(1<<m)-1;
    for(int i=0;i<=MAX;i++)
     if((!(i&(i<<1)))&&(!(i&(i<<2)))) 
     {
       a[++K]=i; 
       num[K]=0;
       int x=i;
       while(x>0)
       {
           num[K]+=(x%2);
           x>>=1;
       }
     }
     
    for(int i=1;i<=n;i++)
     for(int j=1;j<=K;j++)
      for(int k=1;k<=K;k++) dp[i][j][k]=-INF; 
      
    for(int i=1;i<=K;i++)
     if(!(b[1]&a[i])) dp[1][i][1]=num[i];
     
    for(int i=2;i<=n;i++)
     for(int j=1;j<=K;j++)
      if(!(b[i]&a[j]))
      {
          for(int k=1;k<=K;k++)
           if(!(a[j]&a[k]))
           {
           for(int l=1;l<=K;l++)
           {
              if((a[l]&a[k])||(a[l]&a[j])) continue;
              if(dp[i-1][k][l]==-INF) continue;
              dp[i][j][k]=max(dp[i][j][k],dp[i-1][k][l]+num[j]);
           }
         }
      }
      
   int ans=0;
   for(int i=1;i<=K;i++)
    for(int j=1;j<=K;j++) ans=max(ans,dp[n][i][j]);
   printf("%d\n",ans);
   return 0;    
}