POJ1185炮兵陣地狀壓DP

和上題一樣,就是多了一維

輸入要注意,我開始用cin>>t,wa到哭,聽了roll的話之後改了,立馬ac

#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
int n,m;
int map[200];
int num[200];
int dp[200][200][220];
int count(int x)
{
    int sum=0;
    for(int i=0;i<m;i++)
    {
        if(x&(1<<i)) sum++;
    }
    return sum;
}
int main()
{
    int i,j,k;
    char str[250];
    int t;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        memset(map,0,sizeof(map));
        for(i=0;i<n;i++)
        {
            scanf("%s",str);
            for(j=0;j<m;j++)
            {
                if(str[j]=='H') map[i]|=(1<<j);
            }
        }
        int a[10000],cnt=0;
        for(i=0;i<(1<<m);i++)
        {
            if(i&(i<<1)) continue;
            if(i&(i<<2)) continue;
            a[cnt++]=i;
        }
        memset(dp,-1,sizeof(dp));
        for(i=0;i<cnt;i++)
        {
            num[i]=count(a[i]);
            if(map[0]&a[i]) continue;
            dp[0][0][i]=num[i];
        }
        for(i=1;i<n;i++)
        {
            for(t=0;t<cnt;t++)
            {
                if(map[i]&a[t]) continue;
                for(j=0;j<k;j++)
                {
                    if(a[t]&a[j]) continue;
                    for(k=0;k<cnt;k++)
                    {
                        if(a[t]&a[k]) continue;
                        if(dp[i-1][j][k]==-1) continue;
                        dp[i][k][t]=max(dp[i][k][t],dp[i-1][j][k]+num[t]);
                    }
                }
            }
        }
        int ans=0;
        for(i=0;i<cnt;i++)
        {
            for(j=0;j<cnt;j++)
            {
                ans=max(ans,dp[n-1][i][j]);
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}



發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章