POJ1185 狀態壓縮DP 又是一道有意義的水題

之前做過POJ3285，那道題下一行的狀態只和上一行有關，所以每一行只需要記錄一行的狀態。

這一個炮兵陣地，炮可以打兩行，所以不僅僅與上一行有關，還與上上一行有關，所以每一行的狀態都由上兩行狀態推出，因爲要記錄兩個狀態，所以用了一個三維的dp，

dp[i][j][k]，表示第i行狀態爲state[k]，第i-1行狀態爲state[j]時前I行最多有幾個炮兵。

寫的時候一開始在初始化的時候沒有注意硬件條件，掛了一發。

寫完這個程序，100+行，編譯了一下，還是有一個for循環中int忘記定義，除了這個之外就沒有錯誤了。這種感覺真不錯，不過還是有邏輯錯。

第一個邏輯錯是混淆了POJ3285和這個題的所求的東西，在特判行數爲1的時候統計成了此時的狀態數量。

第二個邏輯錯是初始化的時候沒加上對硬件條件的判斷。

真是的，要謹慎啊。

不能疏忽。

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;

int state[62];
int num[62];
int dp[102][62][62];//dp[i][j][k] 表示前i行中 i行爲k種狀態 i-1行爲j種狀態時所能放置的最大炮兵數
int tot=0;
int row[200];
int cal(int i)
{
    int count=0;
    while(i>0)
    {
        count++;
        i-=(i&(-i));
    }
    return count;
}
void init(int n)
{
    int k=1<<n;
    for(int i=0;i<k;i++)
    {
        if((i&(i<<1))==0&&(i&(i<<2))==0)
        {
            num[tot]=cal(i);
            state[tot++]=i;
        }
    }
}

char tmp[100];
int main()
{
    memset(dp,0,sizeof(dp));
    int m,n;
    scanf("%d%d",&m,&n);
    init(n);
    for(int i=0;i<m;i++)
    {
        scanf("%s",tmp);
        for(int j=n-1;j>=0;j--)
        {
            if(tmp[n-1-j]=='P')
                row[i]+=1<<j;
        }
    }
    if(m==1)
    {
        int count=-1;
        for(int i=0;i<tot;i++)
        {
            if((row[0]&state[i])==state[i])
                count=max(num[i],count);
        }
        printf("%d\n",count);
        return 0;
    }
    for(int i=0;i<tot;i++)
    {
        for(int j=0;j<tot;j++)
        {
            if((row[0]&state[i])==state[i]&&(row[1]&state[j])==state[j]&&(state[i]&state[j])==0)
            {
                dp[1][i][j]=num[i]+num[j];
            }
        }
    }
    if(m==2)
    {
        int count=-1;
        for(int i=0;i<tot;i++)
        {
            for(int j=0;j<tot;j++)
            {
                count=max(dp[1][i][j],count);
            }
        }
        printf("%d\n",count);
        return 0;
    }
    for(int i=2;i<m;i++)
    {
        for(int j=0;j<tot;j++)
        {
            if((state[j]&row[i])==state[j])
            {
                for(int k=0;k<tot;k++)
                {
                    for(int l=0;l<tot;l++)
                    {
                        if(dp[i-1][k][l]&&((state[j]&state[k])==0)&&((state[j]&state[l])==0))
                        {
                            dp[i][l][j]=max(dp[i][l][j],dp[i-1][k][l]+num[j]);
                        }
                    }
                }
            }
        }
    }
    int max_=-1;
    for(int i=0;i<tot;i++)
    {
        for(int j=0;j<tot;j++)
        {
            if(dp[m-1][i][j]>max_)
                max_=dp[m-1][i][j];
        }
    }
    printf("%d\n",max_);
    return 0;
}