炮兵阵地 状压dp

题目链接

//dp[x][y][z]:x表示当前的一行，y表示当前行的状态，z表示当前行的上一行状态
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
using namespace std;
int dp[110][100][100],num[100],t,val[200],he[200];
int NUM(int x)
{
    int i,k=0;
    for(i=0; i<=10; i++)
    {
        if(x&(1<<i))
            k++;
    }
    return k;
}
void init()
{
    int i,j,x,y;
    for(i=0; i<=(1<<10); i++)
    {
        x=i>>1;
        y=i>>2;//因为在横向的攻击范围为2，所以需要分别将i右移一位和右移两位保证横向的相邻大炮不受相互攻击。例如i为1001001或101001，1表示大炮的位置，1001001符合条件而101001不符合条件。
        if((!(x&i))&&(!(y&i)))
            {
                he[t]=i;//初始化满足条件的二进制数
                num[t++]=NUM(i);//在符合的i中1的个数（这一排大炮能放的个数）
            }
    }
}
int main()
{
    int n,m;
    t=1;
    init();
    while(scanf("%d %d",&n,&m)!=EOF)
    {
        char mp[200][20];
        int i,j,k,r;
        memset(val,0,sizeof(val));
        memset(dp,0,sizeof(dp));
        for(i=0; i<n; i++)
            scanf("%s",mp[i]);
        for(i=0; i<n; i++)
            for(j=0; j<m; j++)
            {
                if(mp[i][j]=='H')
                    val[i]+=(1<<j);//将山地（1）和平原（0）转化为二进制数进行储存，进行状态压缩
            }
        int res=0;
        if(n==0&&m==0)
        {
            printf("0\n");
            continue;
        }
        for(i=1; i<t; i++)
        {
            if(he[i]>=(1<<m)) break;
            if(!(val[0]&he[i]))//满足条件山地不能放大炮（山地和大炮的二进制中的1不能重合，以下同样位置的地方是相同的含义）
            {

                dp[0][i][1]=num[i];
                res=max(res,dp[0][i][1]);
            }
        }
        if(n==1)
        {
            printf("%d\n",res);
            continue;
        }
        for(i=1; i<t; i++)
        {
             if(he[i]>=(1<<m)) break;
            if(!(val[1]&he[i]))
            {
                for(j=1; j<t; j++)
                {
                    if(he[j]>(1<<m)) break;
                    if(val[0]&he[j])
                        continue;
                    if(he[i]&he[j])//因为大炮在纵向的攻击范围也是2，所以在纵行的满足二进制条件的数的1也不能有重合的（就是1在一个数中的位置相同）
                        continue;
                    dp[1][i][j]=max(dp[1][i][j],dp[0][j][1]+num[i]);
                    res=max(res,dp[1][i][j]);
                }
            }
        }
        if(n==2)
        {
            printf("%d\n",res);
            continue;
        }
        for(r=2; r<n; r++)
        {
            for(i=0; i<t; i++)
            {
                if(he[i]>=(1<<m)) break;
                if(val[r]&he[i])
                    continue;
                for(j=0; j<t; j++)
                {
                    if(he[j]>=(1<<m)) break;
                    if(val[r-1]&he[j])
                        continue;
                    if(he[i]&he[j])
                        continue;
                    for(k=0; k<t; k++)
                    {
                        if(he[k]>=(1<<m)) break;
                        if(val[r-2]&he[k])
                            continue;
                        if(he[k]&he[i]||he[k]&he[j])
                            continue;
                        dp[r][i][j]=max(dp[r][i][j],dp[r-1][j][k]+num[i]);
                        res=max(res,dp[r][i][j]);
                    }
                }
            }
        }
        printf("%d\n",res);
    }
    return 0;
}