POJ 1185 炮兵阵地（状压dp）

http://poj.org/problem?id=1185

题意：

思路：

每一行最多只有10列，所以可以用二进制来表示每一行的状态。

d【i】【j】【k】表示第i行状态为k时，并且上一行状态为j时的最大炮兵数。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<sstream>
#include<vector>
#include<stack>
#include<queue>
#include<cmath>
#include<map>
#include<set>
using namespace std;
typedef long long ll;
typedef pair<int,int> pll;
const int INF = 0x3f3f3f3f;
const int maxn = 20+5;

int n, m;

char s[20];
int st[105];
int tmp[105];
int num[105];
int d[105][105][105];

int calc(int x)
{
    int tmp=0;
    while(x)
    {
        if(x&1)  tmp++;
        x>>=1;
    }
    return tmp;
}

int main()
{
    //freopen("in.txt","r",stdin);
    while(~scanf("%d%d",&n, &m))
    {
        getchar();
        for(int i=1;i<=n;i++)
        {
            scanf("%s",&s);
            st[i]=0;
            for(int j=0;j<m;j++)
            {
                st[i]=(st[i]<<1)|(s[j]=='P');
            }
        }

        int top=0;
        for(int i=0;i<(1<<m);i++)  //记录可行状态
        {
            if(!(i&(i<<1)) && !(i&(i<<2)))
            {
                tmp[top]=i;
                num[top]=calc(i);
                top++;
            }
        }

        int ans=0;
        memset(d,-1,sizeof(d));
        for(int i=0;i<top;i++)  //第一行状态
        {
            if((st[1]|tmp[i])==st[1])   d[1][0][i]=num[i];
        }

        for(int i=2;i<=n;i++)
        {
            for(int j=0;j<top;j++)  //第i行状态
            {
                if((st[i]|tmp[j])!=st[i])  continue;

                for(int k=0;k<top;k++)  //第i-1行状态
                {
                    if((st[i-1]|tmp[k])!=st[i-1])  continue;
                    if(tmp[j]&tmp[k])  continue;  //和第i行有冲突

                    for(int t=0;t<top;t++)  //第i-2行状态
                    {
                        if(i>2)
                        {
                            if((st[i-2]|tmp[t])!=st[i-2])  continue;
                            if(tmp[t]&tmp[k])  continue;
                            if(tmp[t]&tmp[j])  continue;
                        }

                        d[i][k][j]=max(d[i][k][j],d[i-1][t][k]+num[j]);
                    }
                }
            }
        }
        
        for(int i=0;i<top;i++)
            for(int j=0;j<top;j++)
            ans=max(ans,d[n][i][j]);
        printf("%d
", ans);
    }
    return 0;
}