poj 1185 炮兵阵地 状态压缩dp

状态压缩dp 不过这个题目还是有很多要注意的地方。
优化一 先预处理出那些有用的状态。
优化二 循环的时候先判断可行再走下一层


#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn=1e2+9;
int dp[2][maxn][maxn],c[maxn],sum[maxn];
char a[maxn][13];
int n,m,lon;

int check(int tmp)
{
    for(inti=1;i<=m;i++)
    if(tmp&(1<<(i-1)))
        {
            if(tmp&(1<<i))return(0);
            if(tmp&(1<<(i+1)))return(0);
        }
        return(1);
}

int cal(int tmp)
{
    tmp=c[tmp];
    int ret=0;
    for(inti=0;i<m;i++)
    if(tmp&(1<<i))
    ret++;
    return(ret);
}

int cal(int t,int tmp1)
{
    int tmp=c[tmp1];
    int ret=0;
    for(inti=0;i<m;i++)
    if(a[t][i+1]=='H'&&(tmp&(1<<i)))
    return(-1);
    return(sum[tmp1]);
}

int main()
{
    scanf("%d%d",&n,&m);
    for(inti=0;i<1<<m;i++)
    if(check(i))
    c[++lon]=i;

    for(inti=1;i<=lon;i++)
    sum[i]=cal(i);

    for(inti=1;i<=n;i++)
    scanf("%s",&a[i][1]);


    for(inti=1;i<=lon;i++)
    dp[1][i][1]=cal(1,i);

    for(intt=2;t<=n;t++)
    {
        for(int k=1;k<=lon;k++)
        if(cal(t,k)!=-1)
        for(int i=1;i<=lon;i++)
        if(!(c[i]&c[k]))
        for(int j=1;j<=lon;j++)
        if(!(c[j]&c[k]))
        dp[t%2][k][j]=max(dp[t%2][k][j],dp[(t-1)%2][j][i]+sum[k]);
    }
    int ans=0;
    for(inti=1;i<=lon;i++)
    for(intj=1;j<=lon;j++)
    ans=max(dp[n%2][i][j],ans);
    printf("%d\n",ans);
    return 0;
}

你可能感兴趣的:(poj 1185 炮兵阵地 状态压缩dp)