三行相关的棋盘式状态压缩DP
需要存储前两行的情况进行转移
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
const int N = 110;
const int M = 10;
const int K = 1 << M;
vector<int>state;
vector<int>edge[K];
//第一维 当前枚举的行数
//第二维 第i行状态为x
//第三维 第i-1行状态为y
int f[2][K][K];//滚动数组优化空间
int g[N];//g[i]的二进制对应每一列的情况
int cnt[K];//二进制数中1的个数
int n, m;
//三行相关的状压DP
//第i行状态为a
//第i-1行状态为b
//第i-2行状态为c
//地图上也有限制 为H的地方不能放置
bool check(int state)
{
return !(state & state >> 1 || state & state >> 2);
}
int count(int state)
{
int res = 0;
while (state)res += state & 1, state >>= 1;
return res;
}
int main()
{
cin >> n >> m;
for (int i = 1, j = 0; i <= n; i++, j = 0)
for (char c; j < m && cin >> c; j++)
g[i] += (c == 'H') << j;//在列(第j列)对应的二进制位置加1
//预处理合法状态
//最起码在一行之中
//相邻两个1要相隔两块空地
for (int i = 0; i < 1 << m; i++)
if (check(i))
{
state.push_back(i);
cnt[i] = count(i);
}
//分析所有可能的转移方式
for (int cur : state)
for (int pre : state)
if (!(cur & pre))edge[cur].push_back(pre);
int size = state.size();
for (int i = 1; i <= n + 2; i++)
for (int x : state)//第i行状态
if (!(g[i] & x))//第i行的部署情况应满足g[i]表示的地形
for (int y : edge[x])//第i-1行状态
for (int z : edge[y])//第i-2行状态
if(!(x&z))
f[i & 1][x][y] = max(f[i & 1][x][y], f[i - 1 & 1][y][z] + count(x));
//这里是两行一起尝试(同时尝试枚举两行)
//x状态可以由y转移而来
//y状态可以由z转移而来
//计算到n+2行 且n+2行与n+1行部署数皆为0
//即为所求的部署前n行的最大值
cout << f[n + 2 & 1][0][0] << endl;
return 0;
}