洛谷P2704 状态压缩DP

 三行相关的棋盘式状态压缩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;
}