洛谷 P2704 [NOI2001]炮兵阵地 (状压dp)

P2704 [NOI2001]炮兵阵地

洛谷链接

题目描述

司令部的将军们打算在NM的网格地图上部署他们的炮兵部队。一个NM的地图由N行M列组成，地图的每一格可能是山地（用“H” 表示），也可能是平原（用“P”表示），如下图。在每一格平原地形上最多可以布置一支炮兵部队（山地上不能够部署炮兵部队）；一支炮兵部队在地图上的攻击范围如图中黑色区域所示：

如果在地图中的灰色所标识的平原上部署一支炮兵部队，则图中的黑色的网格表示它能够攻击到的区域：沿横向左右各两格，沿纵向上下各两格。图上其它白色网格均攻击不到。从图上可见炮兵的攻击范围不受地形的影响。 现在，将军们规划如何部署炮兵部队，在防止误伤的前提下（保证任何两支炮兵部队之间不能互相攻击，即任何一支炮兵部队都不在其他支炮兵部队的攻击范围内），在整个地图区域内最多能够摆放多少我军的炮兵部队。

输入格式

第一行包含两个由空格分割开的正整数，分别表示N和M；

接下来的N行，每一行含有连续的M个字符（‘P’或者‘H’），中间没有空格。按顺序表示地图中每一行的数据。N≤100；M≤10。

输出格式

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

输入输出样例

输入 #1
5 4
PHPP
PPHH
PPPP
PHPP
PHHP
输出 #1
6

Solution

棋盘类状态压缩dp

限制条件增多，一行的状态受前两行的影响。
每一行的值会受前两行的影响，上面第一行方便枚举，但是再枚举上面第一行不太容易，因为这样我们的值没办法保证互不干涉，于是我们用本行和上一行的排列状态表示一个状态。

令dp[i][j][k]表示i行当前行为j排列上一行为k排列的最大炮兵数量。

然后按照常规方法先除去非法状态，再转移即可。

详细见代码。

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int SZ = 100 + 10;
ll dp[SZ][SZ][SZ]; //dp[i][j][k] 为i行当前行为j上一行为k的最大摆放数量 
int sta[SZ],temp,ph[SZ],onenum[SZ];//ph 为二进制地图 ,sta 可行状态,  
int n,m,M,mp[SZ][15];

inline int getone(int x) //求1的数量
{
	int sum = 0;
	while(x)
	{
		sum ++ ;
		x &= x - 1;
	}
	return sum;
}

int main()
{
	char ch;
	scanf("%d%d",&n,&m);
	for(int i = 1;i <= n;i ++ )
		for(int j = 1;j <= m;j ++ )
		{
			cin >> ch; 
			if(ch == 'H') mp[i][j] = 1;
		}
	for(int i = 1;i <= n;i ++ )
		for(int j = 1;j <= m;j ++ ) 
			ph[i] = (ph[i] << 1) + mp[i][j];  //二进制地图 不可放置区域 
	M = (1 << m) - 1;
	for(int i = 0;i <= M;i ++ )
	{
		if(((i << 1) & i) || ((i << 2) & i)) continue;//同一行中的非法状态 
		sta[ ++ temp] = i;
		onenum[temp] = getone(i);//记一下炮兵数 
	} 
	for(int i = 1;i <= temp;i ++ )
	{
		if(sta[i] & ph[1]) continue;
		dp[1][i][1] = onenum[i]; //初始化第一行,上一行没有,令上一行为摆放法1,即00000..0  
	} 
	for(int i = 1;i <= temp;i ++ ) //初始化第二行 两行为一个状态 
	{
		if(sta[i] & ph[2]) continue;//地形限制 
		for(int j = 1;j <= temp;j ++ )
		{
			if(sta[j] & ph[1]) continue; 
			if(sta[i] & sta[j]) continue; // 相邻的限制 
			dp[2][i][j] = max(dp[1][j][1] + onenum[i],dp[2][i][j]);
		}
	}
	for(int i = 3;i <= n;i ++ )//行 
	{
		for(int j = 1;j <= temp;j ++ )//当前行 
		{
			if(sta[j] & ph[i]) continue; 
			for(int k = 1;k <= temp;k ++ )//上两行 
			{
				if(sta[k] & ph[i - 2]) continue; 
				if(sta[k] & sta[j]) continue;
				for(int l = 1;l <= temp;l ++ )//上一行 
				{
					if(sta[l] & ph[i - 1]) continue; 
					if(sta[l] & sta[j]) continue;
					dp[i][j][l] = max(dp[i][j][l],dp[i - 1][l][k] + onenum[j]);
				}
			}
		}
	}
	ll ans = 0;
	for(int i = 1;i <= temp;i ++ )
		for(int j = 1;j <= temp;j ++ )
		ans = max(ans,dp[n][i][j]); //ans = max(dp[n][i][j])
	printf("%lld\n",ans);
	return 0;
}

2020.4.8