HDU 4539 郑厂长系列故事——排兵布阵 <<状压dp

思路

被这道题折磨死了，只是发上来纪念一下，思路同方格取数(1)，我已经疯了！

代码

#include<bits/stdc++.h>
using namespace std;
int maze[110][15];
int n,m;
vector<int> all[110];
int dp[110][200][200];
int num[1<<10];
bool check(int r,int state)
{
    for(int i=m-1;i>=0;i--)
    {
        if(state&1)
        {
            if(maze[r][i]==0) return false;
            if(i<1) continue;
            if((state>>2)&1) return false;
        }
        state>>=1;
    }
    return true;
}
bool ok(int pre,int now,int f)
{
    if(f==1)
    {
        if((pre>>1)&now||pre&(now>>1)) return false;
        return true;
    }
    else{
        if(!(pre&now)) return true;
        return false;
    }
}
void db()
{
    int cnt=0;
    for(int i=0;i<n;i++)
    {
        all[i].clear();
        for(int state=0;state<1<<m;state++)
        {
            if(check(i,state)) 
            {
                //cout<<i<<":"<<bitset<3>(state)<<endl;
                cnt++,all[i].push_back(state);
                num[state]=__builtin_popcount(state);
            }
        }
    }
    //cout<<cnt<<endl;
}
int main()
{
    //freopen("data.in","r",stdin);
    //freopen("data.out","w",stdout);
    while(~scanf("%d%d",&n,&m))
    {
        memset(maze,0,sizeof(maze));
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<m;j++)
            {
                scanf("%d",&maze[i][j]);
            }
        }
        db();
        memset(dp,0,sizeof(dp));
        int ans=0;
        if(n==1)
        {
            for(int i=0;i<all[0].size();i++)
            {
                ans=max(ans,num[all[0][i]]);
            }
            goto ending;
        }
        for(int i=0;i<all[0].size();i++)
        {
            for(int j=0;j<all[1].size();j++)
            {
                int now=all[1][j],pre=all[0][i];
                if(ok(pre,now,1))
                {
                    dp[1][j][i]=max(num[all[1][j]]+num[all[0][i]],dp[1][j][i]);
                    //cout<<dp[1][j][i]<<endl;
                    ans=max(ans,dp[1][j][i]);
                }
            }
        }
        for(int i=2;i<n;i++)
        {
            for(int j=0;j<all[i].size();j++)
            {
                for(int k=0;k<all[i-1].size();k++)
                {
                    int pre=all[i-1][k],now=all[i][j];
                    if(!ok(pre,now,1)) continue;
                    else{
                        for(int l=0;l<all[i-2].size();l++)
                        {
                            int pp=all[i-2][l];
                            if(ok(pp,now,2)&&ok(pp,pre,1))
                            {
                                dp[i][j][k]=max(dp[i][j][k],dp[i-1][k][l]+num[all[i][j]]);
                                ans=max(ans,dp[i][j][k]);
                            }
                        }
                    }
                }
            }        
        }
        ending:
        printf("%d
",ans);
    }
}