poj 1185 炮兵阵地 [经典状态压缩DP]
题意:略。
思路:由于每个大炮射程为2,所以如果对每一行状态压缩的话,能对它造成影响的就是上面的两行。
这里用dp[row][state1][state2]表示第row行状态为state2,第row-1行状态为state1时最多可以安放多少大炮。
则递推公式为:dp[i][K][J] = max(dp[i-1][L][K] + num[J])。其中num[J]表示状态J的二进制形式里有多少个1。
代码我是参考的别人的,觉得写得很好。
主要有一下几个地方:
1. 在判断一个数二进制形式有多少个1时,用 x & (x - 1) (具体见代码count_one函数)来判断。这种方法的时间复杂度就是x的二进制中1的个数。
假设有一个数二进制形式为1000位,其中只有一个1,则用最普通方法一位一位来数则需要计算1000次,而用该方法就是1次。
2. 在判断一个状态是否合法时(即该状态内不能有两个1距离在2以内),用x & (x << 1),x & (x << 2) (具体见代码ok函数)来判断,这与我一位一位比较的笨方法高下立见。
3. 在判断一个状态在地图中某一行是否合法时(即地图上的'H'处不能放置大炮),将地图的每一行转换成了一个数的二进制形式,'H' 为1,'P'为0。然后用数组line[]将每一行转换成的数字存储起来。之后假设要判断状态s能否放在第i行,则判断line[i] & s是否为0。如果不为0则说明该状态一定在'H'处出现了大炮,是不合法的。
除了上面这些,我在写完之后提交了几次发现wa。
经过检查,发现了原因:
别人的代码中,在求最终结果都是进行完dp后将dp数组遍历一次,求最大值即可。
而我写的则是在dp过程中记录最大值。这思想是没错的,但并没有注意到dp的几重循环是从第二行开始的,而地图第一行的dp值我是在进行dp前单独初始化的。这样子肯定错了,因为当最大值出现在第一行中时,我就记录不到了。后来我在第一行初始化时记录下最大值,又在接下来的dp过程中记录一下,就ac了,但不如直接在dp之后遍历一遍来得简洁,就作罢了。
1 #include<stdio.h> 2 #include<iostream> 3 #include<string.h> 4 #include<algorithm> 5 using namespace std; 6 int n, m, sta[62], dp[120][62][62], tot, line[120], num[62]; 7 char map[105][13]; 8 bool ok(int i)//判断状态i是否合法,即是否有两个1距离小于等于2 9 { 10 if (i & (i<<1)) return 0; 11 if (i & (i<<2)) return 0; 12 return 1; 13 } 14 bool can(int row,int state)//判断状态state是否可以放在地图第row行 15 { 16 if (state & line[row]) return 0; 17 return 1; 18 } 19 int count_one(int x)//计数x的二进制状态有多少1 20 { 21 int res = 0; 22 while (x) 23 { 24 res++; 25 x &= x - 1; 26 } 27 return res; 28 } 29 int getdp() 30 { 31 memset(dp, -1, sizeof(dp)); 32 for (int i = 0; i < tot; i++) 33 { 34 num[i] = count_one(sta[i]); 35 if (can(1, sta[i])) 36 dp[1][0][i] = num[i]; 37 } 38 for (int i = 2; i <= n; i++) 39 for (int j = 0; j < tot; j++) if (can(i, sta[j])) 40 for (int k = 0; k < tot; k++) 41 { 42 if (sta[j] & sta[k]) continue; 43 for (int l = 0; l < tot; l++) 44 { 45 if (sta[j] & sta[l]) continue; 46 if (dp[i-1][l][k] == -1) continue; 47 dp[i][k][j] = max(dp[i][k][j], dp[i-1][l][k] + num[j]); 48 } 49 } 50 int res = 0; 51 for (int i = 1; i <= n; i++) 52 for (int j = 0; j < tot; j++) 53 for (int k = 0; k < tot; k++) 54 res = max(res, dp[i][j][k]); 55 return res; 56 } 57 int main() 58 { 59 //freopen("data.in", "r", stdin); 60 while (~scanf("%d%d",&n, &m) && n && m) 61 { 62 tot = 0; 63 for (int i = 0; i < (1<<m); i++) 64 if (ok(i)) sta[tot++] = i;//预处理所有有效状态 65 memset(line, 0, sizeof(line)); 66 for (int i = 1; i <= n; i++)//将地图每一行的地形转换成二进制 67 { 68 scanf("%s", map[i]); 69 for (int j = 0; j < m; j++) if (map[i][j] == 'H') 70 line[i] += (1<<j); 71 } 72 printf("%d\n", getdp()); 73 } 74 return 0; 75 }