https://loj.ac/problem/10173

题目描述

  一个\(N\times M\)的网格,每个炮兵可以攻击到从它的位置上下左右\(2\)格以内远的位置,有一些位置不能放置炮兵,求部署时炮兵两两间不攻击的方案数。

思路

  我们先无视该位置能否放炮兵,先预处理处如果这一行都能放炮兵时的状态,接下来考虑由于炮兵会影响到两行,暴力记录空间难以开下,所以我们省去堆当前这一行状态的保存,直接记\(f[i][S1][S2]\)表示第\(i\)行时\(i-1\)行为\(S1\)\(i-2\)行为\(S2\)时的答案数,那么我们直接枚举当前行,前两行的状态进行转移即可。

代码

#include<bits/stdc++.h>
using namespace std;

int a[110];
int sum[66],f[110][66][66],p[66];
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        int s=0;
        for(int j=1;j<=m;j++)
        {
            char ch=getchar();
            while(ch!='P'&&ch!='H')ch=getchar();
            s=s*2+(ch=='H');
        }
        a[i]=s;
    }
    int cnt=0;
    for(int i=0;i<(1<<m);i++)
    {
        if((i&(i<<1))||(i&(i<<2)))continue ;
        int k=0;
        for(int j=0;j<m;j++)if(i&(1<<j))++k;
        p[++cnt]=i;
        sum[cnt]=k;
    }
    for(int i=1;i<=cnt;i++)
        if(!(p[i]&a[1]))f[1][i][0]=sum[i];
    for(int i=1;i<=cnt;i++)
        for(int j=1;j<=cnt;j++)
            if(!(p[i]&p[j])&&!(p[i]&a[2]))f[2][i][j]=f[1][j][0]+sum[i];
    for(int i=3;i<=n;i++)
    {
        for(int j=1;j<=cnt;j++)
        {
            if(p[j]&a[i])continue ;
            for(int l=1;l<=cnt;l++)
            {
                if(p[j]&p[l])continue ;
                for(int r=1;r<=cnt;r++)
                    if(!(p[j]&p[r]))f[i][j][l]=max(f[i][j][l],f[i-1][l][r]+sum[j]);
            }
        }
    }
    int ans=0;
    for(int i=1;i<=cnt;i++)
        for(int j=1;j<=cnt;j++)
            ans=max(ans,f[n][i][j]);
    printf("%d",ans);
}
01-07 13:49