状态压缩 HDU4539 郑厂长系列故事——排兵布阵

多组n *m

0不能放1可以放

每个士兵可以攻击到并且只能攻击到与之曼哈顿距离为2的位置以及士兵本身所在的位置。

#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<math.h>
#include<map>

using namespace std;

#define MAXN 1<<10
int z[][];
int x[];
int dp[][][]; //dp[i][j][k] 第i行状态为num[j]第i-1行状态为num[k] 的最大士兵数目
int num[];          //合法状态
int cn[];           //对应可以放的人数目

int Count(int a)
{
    int ans=;
    while(a>)
    {
        if(a&)
            ans++;
        a=a>>;
    }
    return ans;
}
int main()
{
    int n,m;

    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(int i=;i<n;i++)
            for(int j=;j<m;j++)
                scanf("%d",&z[i][j]);
        memset(x,,sizeof(x));
        memset(dp,-,sizeof(dp));
        int cnt=;
        int en=<<m;
        for(int i=;i<en;i++)
            if(((i&(i<<)))==)
            {
                cn[cnt]=Count(i);
                num[cnt++]=i;
            }
        for(int i=;i<n;i++)
        {
            for(int j=;j<m;j++)
                if(!z[i][j])
                    x[i]|=(<<j);
        }

        for(int i=;i<cnt;i++)  //初始化一下
        {
            if(x[]&num[i])
                continue;
            dp[][i][]=cn[i];
        }

        for(int i=;i<n;i++)  //第一行开始转移
        {
            for(int j=;j<cnt;j++)  //这行的状态
            {
                if(x[i]&num[j]) //这一行和这个状态不能矛盾
                    continue;
                for(int k=;k<cnt;k++) //上一行的状态
                {
                    if((num[j]&(num[k]<<))||(num[j]&(num[k]>>)))//这一行状态和上一行左 右
                        continue;
                    for(int k1=;k1<cnt;k1++) //上面第二行状态
                    {
                        if(num[j]&num[k1])   //上一行和上上行
                            continue;
                        if(num[k]&(num[k1]<<))
                            continue;
                        if(num[k]&(num[k1]>>))
                            continue;
                        if(dp[i-][k][k1]==-)
                            continue;
                        dp[i][j][k]=max(dp[i][j][k],dp[i-][k][k1]+cn[j]);
                    }
                }
            }
        }
        int ans=;
        for(int i=;i<cnt;i++)
            for(int j=;j<cnt;j++)
                ans=max(dp[n-][i][j],ans);
        printf("%d\n",ans);
    }
    return ;
}