POJ1185炮兵阵地（状态压缩DP)

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int MAX = 110;
const int INF = 0x3f3f3f3f;
const int maxn = 70;
///state[i][j]表示第i行第j个合法状态，one表示第i行第j个合法状态中含1的个数
int state[MAX][MAX],one[MAX][MAX],ans;
///stnum[i]表示i行合法的状态数，sum[i]为i状态下1的个数
int stsum[MAX],sum[MAX*20];
///dp[i][j][k]表示第i行第j个状态第-1行第k个状态含有的最多1的个数
int dp[MAX][maxn][maxn],map[MAX][MAX],n,m;
void init( )
{
    ans=0;
    memset(dp,0,sizeof(dp));
    memset(map,0,sizeof(map));
    memset(one,0,sizeof(one));
    memset(stsum,0,sizeof(stsum));
}
//每个状态里面含有1的数量算出来
void onesum( )
{
    for(int i=0 ; i<=(1<<10) ; i++)
    {
        int t=0;
        for(int j=0 ; j<=10 ; ++j)
        if(i&(1<<j))
        t++;
        sum[i]=t;
    }
}
///判断状态是否符合
bool ok(int x)
{
    if(x>1&&(x&(x>>1)))
    return 0;
    if(x>2&&(x&(x>>2)))
    return 0;
    return 1;
}
///把第x行中合法的状态全部找出来，存到state数组中，tot是本行所有的p点压缩起来的一个状态
void STATE(int x,int tot)
{
    for(int i=0 ; i<(1<<m) ; i++)
    if(ok(i)&&(i&tot)==i)
    ///（i&tot） == i表示集合i是集合tot的子集合，意思是i里面的含有的列都是p点
    {
        stsum[x]++;
        int t=stsum[x];
        state[x][t]=i;
        one[x][t]=sum[i];
    }
}
int main( )
{
    onesum( );
    char tp[MAX];
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        init( );
        stsum[0]=maxn;
        for(int i=1 ; i<=n ; i++)
        {
            scanf("%s",tp);
            int k=0;
            for(int j=0 ; j<m ; j++)
            {   //将图变为0，1图
                 map[i][j] = (tp[j] == ‘P‘ ? 1 : 0);
                k += (map[i][j] ? (1 << j) : 0);
            }
            STATE(i,k);

        }
        ///第一行
        for(int i=1 ; i<=stsum[1] ; i++)
        for(int j=1 ; j<=stsum[0] ; j++)
        dp[1][i][j]=one[1][i];
        for(int i=2 ; i<=n ; i++)
        for(int j=1 ; j<=stsum[i] ; j++)///枚举第i行的状态
        for(int k=1 ; k<=stsum[i-1] ; k++)///枚举第i-1行的状态
        {/// 判断两个状态是否有冲突
            if(state[i][j]&state[i-1][k])
            continue;
            int t=0;
            for(int s=0 ; s<=stsum[i-2] ; s++)///枚举第i-2行的状态
            {
                if(state[i][j]&state[i-2][s])/// 判断三个状态是否有冲突
                continue;
                if(state[i-1][k]&state[i-2][s])
                continue;
                t=max(dp[i-1][k][s],t);
            }
            dp[i][j][k]=t+one[i][j];
        }
        for(int j=1 ; j<=stsum[n] ; j++)
        for(int k=1 ; k<=stsum[n-1] ; k++)
        ans=max(ans,dp[n][j][k]);
        printf("%d\n",ans);
    }
    return 0;
}