状压DP [POJ 1185] 炮兵阵地

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
#define N 101
#define M 60

int n,m;
int a[N];
int num[M];
int ok[M],tot;
int dp[N][M][M];        //dp[i][j][k]表示第i行状态为k时、第i-1状态为j时的最大摆放数量

void init()
{
    tot=0;
    memset(a,0,sizeof(a));
    memset(dp,0,sizeof(dp)); 

    int MAX=1<<m;
    for(int i=0;i<MAX;i++)
    {
        if((i&(i<<1)) || (i&(i<<2))) continue;
        ok[tot]=i;
        int tmp=i;
        int cnt=0;
        while(tmp)
        {
            cnt=tmp%2==1?cnt+1:cnt;
            tmp/=2;
        }
        num[tot]=cnt;
        tot++;
    }
}
void solve()
{
    int i,j,l,k;
    int MAX=1<<m;
    for(i=1;i<=n;i++)
    {
        for(j=0;j<tot;j++)
        {
            for(k=0;k<tot;k++)
            {
                if((a[i]&ok[k])!=ok[k]) continue;
                if(ok[k]&ok[j]) continue;

                for(l=0;l<tot;l++)
                {
                    if(ok[l]&ok[k] || ok[l]&ok[j] ) continue;
                    dp[i][j][k]=max(dp[i][j][k],dp[i-1][l][j]+num[k]); 
                }
            }
        }
    }
    int res=0;
    for(i=n,j=0;j<tot;j++) 
    {
        for(k=0;k<tot;k++)
        {
            res=max(res,dp[i][j][k]);
        }
    }
    printf("%d\n",res);
}
int main()
{
    char ch;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        init();
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=m;j++)
            {
                scanf(" %c",&ch);
                int x=ch==‘P‘?1:0;
                a[i]=(a[i]<<1)+x;
            }
        }
        solve();
    }
    return 0;
}