状压dp(二)——#10173. 「一本通 5.4 练习 2」炮兵阵地

题目链接:https://loj.ac/problem/10173
解题思路
本题也与国王类似,只不过他需要看前面两行的状态,因此dp需要记本行与前一行的状态,才能转移,否则,如果只记录本行的状态,会导致错误。因为前面两个dp的状态可能不能同时取最大。
AC代码

#include <iostream>
#include <stdio.h>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
int n,m,cnt;
char mp[105][15];
int dp[105][105][105];
int sta[105];
vector<int>v[105];
vector<int>s[105];
bool check(int a,int b)
{
    for(int i=0;i<m;++i)
    {
        if(mp[b][i]=='H'&&((1<<(m-i-1))&a))
        return false;
    }
    return true;
}
int ct(int a)
{
    int tmp=0;
    while(a)
    {
        if(a&1)
        tmp++;
        a>>=1;
    }
    return tmp;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;++i)
    scanf("%s",&mp[i]);
    dp[0][0][0]=0;
    for(int i=0;i<(1<<m);++i)
    {
        if(!(i&(i>>1))&&!(i&(i>>2)))
        {
            for(int j=1;j<=n;++j)
            {
                if(check(i,j))
                {
                    v[j+1].push_back(i);
                    int ss=ct(i);
                    s[j+1].push_back(ss);
                }
            }
        }
    }
    v[0].push_back(0);
    v[1].push_back(0);
    for(int i=2;i<=n+1;++i)
    for(int j=0;j<v[i].size();++j)
    for(int k=0;k<v[i-1].size();++k)
    for(int t=0;t<v[i-2].size();++t)
    if(!(v[i][j]&v[i-1][k])&&!(v[i][j]&v[i-2][t])&&!(v[i-1][k]&v[i-2][t]))
    dp[i-1][j][k]=max(dp[i-1][j][k],dp[i-2][k][t]+s[i][j]);
    int ans=0;
    for(int i=0;i<v[n].size();++i)
    for(int j=0;j<v[n+1].size();++j)
    ans=max(ans,dp[n][j][i]);
    printf("%d\n",ans);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_44491423/article/details/108281866