CEOI 2002 Bugs公司 - 状态压缩DP

黑书P138 例题10

#include<stdio.h>
#include<string.h>
#include<math.h>
int n,m;
int maps[155][15];
int f[2][60006];
int a[15];
int p[15];
 
int max(int x,int y)
{
    if (x>y)
        return x;
    return y;
}
void solve(int i,int j,int last,int now,int num)
{
    int x,y;
    if (j>m)
    {
        f[i&1][now]=max(f[i&1][now],f[(i-1)&1][last]+num);
        return ;
    }
    if (maps[i][j]==0)
        x=2;
    else if (p[j]>0)
        x=p[j]-1;
    else
        x=0;
    y=now+x*a[j];
    solve(i,j+1,last,y,num); //当前不放置，状态为y的决策
 
    if (j+1<=m && !p[j] && !p[j+1] && maps[i][j] && maps[i][j+1] && i>=3)
    {
        y=now+2*a[j]+2*a[j+1];
        solve(i,j+2,last,y,num+1); //放置高3,宽2的进去。
    }
    if (j+2<=m && p[j]<=1 && p[j+1]<=1 && p[j+2]<=1 && maps[i][j] && maps[i][j+1] && maps[i][j+2])
    {
        y=now+2*a[j]+2*a[j+1]+2*a[j+2];
        solve(i,j+3,last,y,num+1);//放置高2,宽3的进去。
    }
    return ;
}
int main()
{
    char s[15];
    a[1]=1;
    for (int i=2;i<=12;i++)
        a[i]=a[i-1]*3;
    while (scanf("%d%d",&n,&m)==2)
    {
        for (int i=1;i<=n;i++)
        {
            scanf("%s",s);
            for (int j=1;j<=m;j++)
                if (s[j-1]=='.')
                    maps[i][j]=1;
                else
                    maps[i][j]=0;
        }
        memset(f,-1,sizeof(f));
        f[0][a[m+1]-1]=0;
        for (int i=1;i<=n;i++)
        {
            for (int j=0;j<a[m+1];j++)
                f[i&1][j]=-1;
            for (int j=0;j<a[m+1];j++)
            {
                if (f[(i-1) & 1][j]>-1)
                {
                    int tmp=j;
                    for (int k=1;k<=m;k++)
                    {
                        p[k]=tmp % 3;
                        tmp/=3;
                    }
                    solve(i,1,j,0,0);
                }
            }
        }
        int ans=0;
        for (int i=0;i<a[m+1];i++)
            ans=max(ans,f[n & 1][i]);
        printf("%d\n",ans);
    }
    return 0;
}