POJ 1185 狀態壓縮DP

當前行的狀態與前兩行的狀態相關，預處理出可行的狀態。

狀態方程: dp[i][k][t] = max(dp[i][k][t] ,dp[i-1][j][k]+num[t]) 

dp[i][j][k] 表示第i行的狀態爲k,第i-1行狀態爲j時能放大炮個數的最大值

0-(1<<10)可行的狀態總數不超過60

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <string>
#include <iomanip>
#include <queue>
#include <vector>
#include <set>
#include <map>
typedef long long LL;
#define inf 0x3f3f3f3f
#define P system("pause")
using namespace std;
#define M 105
int dp[M][60][60],m,n;
char mp[M][15];
int s[60],ssize,cur[M],num[60];//s爲枚舉的可行狀態總數，num表示狀態爲t時的大炮數
bool isok(int status,int k)
{
    if(cur[k]&status) //狀態status爲1的地方cur不能爲1，否則衝突
        return 0;
    return 1;
}
int count(int x) //x化爲二進制後含有的1的個數
{
    int s=0;
    while(x)
    {
        if(x&1)
            s++;
        x>>=1;
    }
    return s;
}
int judge(int x)//距離<=2都衝突
{
    if(x&(x<<1))return 0;
    if(x&(x<<2))return 0;
    if(x&(x>>1))return 0;
    if(x&(x>>2))return 0;
    return 1;
}
void init()
{
    int tot=1<<m;
    ssize=0;
    for(int i=0;i<tot;i++)
    {
        if(judge(i))
            s[ssize++]=i;
    }
}
int main()
{
    int i,j,k,t;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        init();
        for(i=1;i<=n;i++)
            scanf("%s",mp[i]+1);
        for(i=1;i<=n;i++)
        {
            cur[i]=0;
            for(j=1;j<=m;j++)
            {
                if(mp[i][j]=='H')
                    cur[i]+=(1<<(j-1)); //H看做1，表示不能放
            }
        }
        memset(dp,-1,sizeof(dp));
        for(i=0;i<ssize;i++)
        {
            num[i]=count(s[i]);
            if(isok(s[i],1))
                dp[1][0][i]=num[i];
        }
        for(i=2;i<=n;i++)
        {
            for(t=0;t<ssize;t++)
            {
                if(!isok(s[t],i))continue;
                for(k=0;k<ssize;k++)
                {
                    if(s[k]&s[t])continue;
                    for(j=0;j<ssize;j++)
                    {
                        if(s[t]&s[j])continue;
                        if(dp[i-1][j][k]==-1)continue;
                        dp[i][k][t]=max(dp[i][k][t],dp[i-1][j][k]+num[t]);
                    }
                }
            }
        }
        int ans=0;
        for(i=1;i<=n;i++)
            for(j=0;j<ssize;j++)
                for(k=0;k<ssize;k++)
                    ans=max(ans,dp[i][j][k]);
        printf("%d\n",ans);
    }
    return 0;
}