poj1185 炮兵阵地(状压dp)

这题和郑长——派兵布阵相似,同理按行dp,炮兵的范围和那个一样都涉及了两行,于是设计状态也是一样的
#include<iostream>
#include<math.h>
#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;
#define oo 0x3f3f3f3f
int dp[102][70][70];
int st[70],sum[70],all;
int Map[102];

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

void GetStatus()
{
    all=0;
    for(int i=0;i<=1<<10;i++)
    {
        if(!(i&(i<<2))&&!(i&(i<<1)))
        {
            st[all]=i;
            sum[all]=Sum(i);
            all++;
        }
    }
    //cout<<all<<endl;
}

void TheMap(int n,int m)
{
    char val;
    for(int i=1;i<=n;i++)
    {
        Map[i]=0;
        getchar();
        for(int j=0;j<m;j++)
        {
            scanf("%c",&val);
            if(val=='H')
                Map[i]|=(1<<j);
        }
    }
}

int Dp(int n,int m)
{
    memset(dp,-1,sizeof dp);
    int now=1<<m,ans=0;
    for(int i=0;st[i]<now;i++)
        if(!(st[i]&Map[1]))
            dp[1][i][0]=sum[i];
    for(int i=2;i<=n;i++)
    {
        for(int j=0;st[j]<now;j++)
        {
            if(st[j]&Map[i]) continue;
            for(int p=0;st[p]<now;p++)
            {
                if(st[j]&st[p]) continue;
                for(int pp=0;st[pp]<now;pp++)
                {
                    if((st[j]&st[pp])||(st[p]&st[pp])) continue;
                    if(dp[i-1][p][pp]==-1) continue;
                    dp[i][j][p]=max(dp[i][j][p],dp[i-1][p][pp]+sum[j]);
                }
            }
        }
    }
    for(int i=0;st[i]<now;i++)
        for(int j=0;st[j]<now;j++)
            ans=max(ans,dp[n][j][i]);
    return ans;
}

int main()
{
    int n,m,ans;
    int val;
    GetStatus();
    while(scanf("%d %d",&n,&m)!=EOF)
    {
        TheMap(n,m);
        ans=Dp(n,m);
        printf("%d\n",ans);
    }
    return 0;
}










你可能感兴趣的:(dp,poj)