【POJ】1185 炮兵阵地

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define MAXN 110
#define MAXM 66
char s[MAXN][MAXM];
int n,m,size;
int dp[MAXN][MAXM][MAXM];
int h[MAXN],a[MAXN],cnt[MAXN];
inline bool OK(int x)
{
    if(x&(x<<1))
        return false;
    if(x&(x<<2))
        return false;
    return true;
}
int One(int x)
{
    int ans;
    for(ans=0;x;x>>=1)
    {
        if(x&1)
            ans++;
    }
    return ans;
}
void Init()
{
    int i;
    for(i=size=0;i<(1<<m);i++)
    {
        if(OK(i))
        {
            a[size]=i;
            cnt[size++]=One(i);
        }
    }
}
inline bool Unfit(int x,int y)
{
    return h[x]&a[y];
}
inline bool Two(int x,int y)
{
    return a[x]&a[y];
}
void DoIt()
{
    int i,j,k,t,ans;
    memset(dp,0,sizeof(dp));
    for(i=0;i<size;i++)
    {
        if(!Unfit(1,i))
            dp[1][i][0]=cnt[i];
    }
    for(i=2;i<=n;i++)
    {
        for(j=0;j<size;j++)
        {
            if(Unfit(i,j))
                continue;
            for(k=0;k<size;k++)
            {
                if(Unfit(i-1,k)||Two(j,k))
                    continue;
                for(t=0;t<size;t++)
                {
                    if(Unfit(i-2,t)||Two(t,k)||Two(t,j))
                        continue;
                    dp[i][j][k]=max(dp[i][j][k],dp[i-1][k][t]+cnt[j]);
                }
            }
        }
    }
    for(ans=0,i=1;i<=n;i++)
    {
        for(j=0;j<size;j++)
        {
            for(k=0;k<size;k++)
                ans=max(ans,dp[i][j][k]);
        }
    }
    printf("%d\n",ans);
}
int main()
{
    int i,j;
    while(~scanf("%d%d",&n,&m))
    {
        for(i=1;i<=n;i++)
        {
            h[i]=0;
            for(j=1;j<=m;j++)
            {
                scanf(" %c",&s[i][j]);
                if(s[i][j]=='H')
                    h[i]+=1<<(j-1);
            }
        }
        Init();
        DoIt();
    }
    return 0;
}