hdu 4539 郑厂长系列故事——排兵布阵 （状压DP）

同 poj 1185 

只是多加一些判断了

就是相邻的行之间 要多加一个哈弗曼为2 的判断

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;

int dp[100][200][200],st[200],sum[200],a[100];
int n,m,cnt;

int get(int x)//处理出x状态下  能放下的炮台数  就是二进制下的1的数量
{
    int s=0;
    while(x)
    {
        if(x&1)s++;
        x>>=1;
    }
    return s;
}
bool J(int x,int y)
{
    if(x&(y<<1))return 0;
    if(x&(y>>1))return 0;
    return 1;
}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        cnt=0;
        for(int i=0;i<n;i++)
        {
            a[i]=0;
            int sbt;
            for(int j=0;j<m;j++)
            {
                scanf("%d",&sbt);
                if(!sbt)a[i]|=(1<<j);
            }
        }

        memset(dp,-1,sizeof dp);

        int tmp=1<<m;
        for(int i=0;i<tmp;i++)//拿出有用的状态
        {
            if((!(i&(i<<2))))
            {
                st[cnt]=i;
                sum[cnt++]=get(i);
            }
        }

        for(int i=0;i<cnt;i++)//处理第一行
        {
            if(!(a[0]&st[i]))
            {
                dp[0][0][i]=sum[i];
            }
        }

        for(int i=1;i<n;i++)
        {
            for(int j=0;j<cnt;j++)//枚举当前行的状态
            {
                if(!(st[j]&a[i]))//如果这个状态和地图不冲突
                {
                    for(int k=0;k<cnt;k++)//枚举i-1行的状态
                    {
                        if(J(st[k],st[j]))//如果i-1行的状态和i行的状态不冲突
                        {
                            for(int p=0;p<cnt;p++)//i-2
                            {
                                if(!(st[j]&st[p]) && J(st[p],st[k]))//如果i-2行的状态和第i行的状态不冲突
                                {
                                    if(dp[i-1][p][k]!=-1)//如果之前的状态合法才能转移
                                    {
                                        dp[i][k][j]=max(dp[i][k][j],dp[i-1][p][k]+sum[j]);
                                    }
                                }
                            }
                        }
                    }
                }
            }
        }

        int ans=0;
        for(int i=0;i<cnt;i++)
        for(int j=0;j<cnt;j++)
        ans=max(ans,dp[n-1][i][j]);

        printf("%d\n",ans);
    }
    return 0;
}