POJ 1185 炮兵布阵(状压dp)

最新推荐文章于 2021-08-05 21:13:03 发布

Sqwlly

最新推荐文章于 2021-08-05 21:13:03 发布

阅读量233

点赞数

分类专栏：算法 ACM POJ 动态规划文章标签：状压dp poj1185

本文链接：https://blog.csdn.net/Eternally831143/article/details/83348976

版权

算法同时被 3 个专栏收录

203 篇文章 1 订阅

订阅专栏

ACM

127 篇文章 0 订阅

订阅专栏

动态规划

48 篇文章 0 订阅

订阅专栏

炮兵布阵

题解：先考虑一维的情况， $d p [i]$ 前 $i$ 行所最多能摆放的炮兵部队的数量，显然无法满足无后效性，无法递推。再考虑加一维， $d p [i] [j]$ 表示第 $i$ 行布局为 $j$ 的情况下所最多能摆放的炮兵部队的数量，此时依然无法递推，因为每一个炮兵影响的范围是前后两格和左右两格，所以在第 $i - 1$ 行布局为 $k$ 的时候无法分辨与布局 $j$ 是否相容。因此考虑再加一维， $d p [i] [j] [k]$ 表示第 $i$ 行布局为 $j$ 第 $i - 1$ 行布局为 $k$ 时最多能摆放的炮兵部队的数量。 $dp[i][j][k] = max\{dp[i-1][k][m]+num[j]\}$ 此时布局 $j$ 和 $k$ 必须相容，否则 $d p [i] [j] [k] = 0$ 。 $n u m [j]$ 表示布局为 $j$ 中的炮兵数量。我们发现 $j$ 和 $m$ 必须相容， $k$ 和 $m$ 必须相容。发现可以满足无后效性。
至于边界即 $\begin{cases} dp[1][j][1]=num[j]\\\\ dp[1][i][j] = max\{dp[1][j][1]+num[j] \} \end{cases}$

代码

#include<cstdio>
#include<cstring>
#define max(a,b) ((a) > (b) ? (a) : (b))
using namespace std;

int n,m;
char map[110][20], num[110], top;
int state[70], cur[70], dp[110][70][70];

bool ok(int x) {
	if((x & (x << 1)) || (x & (x << 2))) return 0;
	return 1;
}

int getNum(int x) {
	int cnt = 0;
	while(x) {
		cnt++;
		x &= (x - 1);
	}
	return cnt;
}

void init() {
	top = 0;
	for(int i = 0; i < (1 << m); ++i) 
		if(ok(i))  state[++top] = i;
	memset(dp,-1,sizeof dp);
}

int main()
{
#ifndef ONLINE_JUDGE
    freopen("input.in","r",stdin);
#endif
	scanf("%d%d",&n,&m);
	init();
	for(int i = 1; i <= n; ++i)
		scanf("%s",map[i] + 1);
	for(int i = 1; i <= n; ++i) {
		cur[i] = 0;
		for(int j = 1; j <= m; ++j){
			if(map[i][j] == 'H') cur[i] |= (1 << (j - 1));
		}
	}
	for(int j = 1; j <= top; ++j) {
		num[j] = getNum(state[j]);
		if((state[j] & cur[1]) == 0) 
			dp[1][j][1] = num[j];
	}
	for(int i = 2; i <= n; ++i) { //第i行
		for(int j = 1; j <= top; ++j) { //第i行布局为j
			if(state[j] & cur[i]) continue;
			for(int k = 1; k <= top; ++k) { //第i-1行布局为k
				if(state[j] & state[k]) continue;
				for(int m = 1; m <= top; ++m) {
					if(state[j] & state[m]) continue; // j与m相容
					if(state[k] & state[m]) continue; // k与m相容
					if(dp[i - 1][k][m] == -1) continue; 
					dp[i][j][k] = max(dp[i][j][k], dp[i - 1][k][m] + num[j]);
				}
			}
		}
	}
	int ans = 0;
	for(int i = 1; i <= top; ++i)
		for(int j = 1; j <= top; ++j) ans = max(ans, dp[n][i][j]);
	printf("%d\n",ans);
    return 0;
}