POJ 1185 炮兵阵地(状压DP)

来源:互联网 发布:单片机时序图 编辑:程序博客网 时间:2024/06/02 02:01

题目链接:点击打开链接

思路:

很容易想到把地图转化为0 1矩阵,从而对每行再用到状态压缩,山用1表示,空地用0表示,即把地图也用状态压缩的方式存起来,便于以后处理

开始先根据一行中每个炮的影响范围为左右各两个距离,枚举出对于行来说,满足要求的排放状态。

然后枚举每行的状态时,先判断此状态和原始地图的二进制表示冲不冲突,因为用1代表的H(山地),加上只能空地放炮,所以把状态和本行二进制表示做按位与,若不为1,则满足要求。

采用自上而下处理,把一个炮的攻击看作前两行,所以对于第i行来讲,i-1行和i-2行与之有影响,以上其他的都没有影响了,我们处理到第i行时,每取到一个合法状态,再去枚举i-1行和i-2行的合法状态,并判断这三个状态之间满不满足要求。

状态方程:

dp[r][i]=max { dp[r-1][j]+dp[r-2][k] } + soldier[i],由于要满足con[i],con[j],con[k]不能有冲突,而且不光dp[r][i]受到dp[r-1][j]和dp[r-2][k]影响,dp[r-1][j]也受到dp[r-2][k]影响,这个转移方程计算起来不容易,所以引入三维的形式 dp[r][i][j]= max{ dp[r-1][j][k]} + soldier[i] } , dp[r][i][j]表示第r行状态为con[i],第r-1行状态为con[j]时的最大值。

// POJ 1185 炮兵阵地.cpp 运行/限制:219ms/2000ms#include <cstdio>#include <cstring>#include <algorithm>#include <iostream>using namespace std;int n, m, map[105];int dp[105][70][70];//dp[i][j][k]表示第i行为状态con[j],第i - 1行为状态con[k]时,能摆放的最大炮兵数int cnt, con[70], soldier[70];//状态,该状态对应的炮兵数(即状态的二进制表示中的1的个数)void init() {//先保存考虑一行时,符合要求的状态和对应炮兵数cnt = 0;int upper = 1 << 10;for (int i = 0; i < upper; i++) {if (i & (i << 1) || i & (i << 2)) continue;int te = i;soldier[cnt] = 0;while (te) {if (te & 1) {soldier[cnt]++;}te >>= 1;}con[cnt++] = i;}//cnt 60}int main(){int upper;init();while (scanf("%d%d", &n, &m) != EOF) {upper = 1 << m;memset(dp, 0, sizeof(dp));for (int i = 0; i < n; i++) {getchar();char te;int num = 0;for (int j = 0; j < m; j++) {scanf("%c", &te);num |= ((te == 'H') << j);}map[i] = num;}//第0行for (int i = 0; i < cnt && con[i] < upper; i++) {if (con[i] & map[0]) continue;dp[0][i][0] = soldier[i];}//第1行for (int i = 0; i < cnt && con[i] < upper; i++) {if (con[i] & map[1]) continue;for (int j = 0; j < cnt && con[j] < upper; j++) {if (con[j] & map[0]) continue;if (con[i] & con[j]) continue;dp[1][i][j] = max(dp[1][i][j], dp[0][j][0] + soldier[i]);}}//2~n - 1行for (int i = 2; i < n; i++) {for (int j = 0; j < cnt && con[j] < upper; j++) {//r行状态if (con[j] & map[i]) continue;for (int t = 0; t < cnt && con[t] < upper; t++) {//r - 1行状态if (con[t] & map[i - 1]) continue;if (con[j] & con[t]) continue;for (int w = 0; w < cnt && con[w] < upper; w++) {//r - 2行if (con[w] & map[i - 2]) continue;if (con[t] & con[w]) continue;if (con[j] & con[w]) continue;dp[i][j][t] = max(dp[i][j][t], dp[i - 1][t][w] + soldier[j]);}}}}//求结果int re = 0;for (int i = 0; i < cnt && con[i] < upper; i++) {for (int j = 0; j < cnt && con[j] < upper; j++) {re = max(re, dp[n - 1][i][j]);}}cout << re << endl;}    return 0;}



原创粉丝点击