洛谷P2704 炮兵阵地 状压dp

题目描述

司令部的将军们打算在 N\times MN×M 的网格地图上部署他们的炮兵部队。

一个 N\times MN×M 的地图由 NN 行 MM 列组成，地图的每一格可能是山地（用 \texttt{H}H 表示），也可能是平原（用 \texttt{P}P 表示），如下图。

在每一格平原地形上最多可以布置一支炮兵部队（山地上不能够部署炮兵部队）；一支炮兵部队在地图上的攻击范围如图中黑色区域所示：

如果在地图中的灰色所标识的平原上部署一支炮兵部队，则图中的黑色的网格表示它能够攻击到的区域：沿横向左右各两格，沿纵向上下各两格。

图上其它白色网格均攻击不到。从图上可见炮兵的攻击范围不受地形的影响。

现在，将军们规划如何部署炮兵部队，在防止误伤的前提下（保证任何两支炮兵部队之间不能互相攻击，即任何一支炮兵部队都不在其他支炮兵部队的攻击范围内），在整个地图区域内最多能够摆放多少我军的炮兵部队。

输入格式

第一行包含两个由空格分割开的正整数，分别表示 NN 和 MM。

接下来的 NN 行，每一行含有连续的 MM 个字符，按顺序表示地图中每一行的数据。

输出格式

一行一个整数，表示最多能摆放的炮兵部队的数量。

输入输出样例

输入 #1复制

5 4
PHPP
PPHH
PPPP
PHPP
PHHP

输出 #1复制

6

说明/提示

对于 100\%100% 的数据，N\le 100N≤100，M\le 10M≤10，保证字符仅包含 p 与 h。

#include <bits/stdc++.h>
using namespace std;
int dp[2][1100][1100],bits[110];
int cnt(int x)//计算1的个数
{
	int ans = 0; 
	while(x){
		if(x & 1) ans++;
		x >>= 1;
	}
	return ans;
}
int main()
{
	char ch;
	int n,m;
	cin >> n >> m;
	for(int i = 1;i <= n; i++){//读入处理有1的表示不能放
		for(int j = 0;j < m; j++){
			cin >> ch;
			if(ch == 'H'){
				bits[i] += (1 << j);
			}
		}
	}
	vector<int> st;
	for(int i = 0;i < (1 << m); i++){//对于一行 选出符合条件的状态 即两两直接的距离大于等于二
		int flag = 1;
		for(int j = 0;j < m; j++){
			if(((i >> j & 1) && ((i >> (j+1) & 1) | (i >> (j+2) & 1)))){
				flag = 0;
				break;
			}
		}
		if(flag) st.push_back(i);
	}
	int si = st.size();
	for(int i = 1;i <= n; i++){//第几行
		for(int j = 0;j < si; j++){//上一行状态
			for(int k = 0;k < si; k++){//当前行状态
				for(int l = 0;l < si; l++){//上上行状态
					int x = st[j],y = st[k],z = st[l];
					if((x & y) | (y & z) | (x & z)) continue;
					if((y & bits[i]) | (x & bits[i-1])) continue;
					dp[i&1][j][k] = max(dp[i&1][j][k],dp[(i-1)&1][l][j]+cnt(y));
				}
			}
		}
	}
	int res = 0;
	for(int i = 0;i < si; i++){
		for(int j = 0;j < si; j++){
			res = max(res,dp[n&1][i][j]);
		}
	}
	cout << res;
	return 0;
}