POJ 1185 炮兵阵地

状态压缩专题第一题，自己想了很久，最终还是以别人的代码为模板写的。

dp共三维，一维是行数，一维是前一行状态，一维是前第二行状态。

PS：直接开三维太大，用s数组记录下所有可能出现的情况，大大减少时间和空间。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<iostream>
using namespace std;
char b[105][15];
int a[105];
int s[105];
int c[105];
int n,m;
int dp[105][105][105];
int Cout (int x)
{
    int ans=0;
    while(x)
    {
        ans+=x&1;
        x>>=1;
    }
    return ans;
}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(int i=0;i<n;i++)
            scanf("%s",b[i]);
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<m;j++)
            {
                if(b[i][j]=='P')a[i]<<=1;
                else a[i]=a[i]<<1|1;
            }
        }
        int ptr=0;//记录下一行中可能出现的摆放情况
        for(int i=0;i<(1<<m);i++)
        {
            if((i&i<<1)||(i&i<<2))continue;//出现连续两个摆放炮台的跳过
            s[ptr]=i;
            c[ptr++]=Cout(i);
        }
        memset(dp,-1,sizeof(dp));
        for(int i=0;i<ptr;i++)
        {
            if(s[i]&a[0])continue;
            dp[0][i][0]=c[i];
        }
        for(int i=1;i<n;i++)
        {
            for(int j=0;j<ptr;j++)
            {
                if(s[j]&a[i])continue;
                for(int k=0;k<ptr;k++)
                {
                    if(s[j]&s[k])continue;//当前行与前一行
                    for(int l=0;l<ptr;l++)
                    {
                        if(s[l]&s[j])continue;//当前行与前第二行
                        if(dp[i-1][k][l]==-1)continue;
                        dp[i][j][k]=max(dp[i][j][k],dp[i-1][k][l]+c[j]);//前一行转移到当前行；
                    }
                }
            }
        }
        int ans=0;
        for(int i=0;i<ptr;i++)
            for(int j=0;j<ptr;j++)
            ans=max(ans,dp[n-1][i][j]);
        printf("%d\n",ans);
    }
    return 0;
}