PKU 1185 炮兵阵地 经典状态压缩dp

来源:互联网 发布:怎么删除淘宝空间图片 编辑:程序博客网 时间:2024/05/29 12:55

思路:状态压缩DP。很经典的状态压缩DP。用int型来表示每行的状态(如果int型的二进制的第i位为1,则表示这一行的第i列有安装大炮)。这样的话由于最多有10列,故由计算可得最多有60种状态。DP部分:dp[r][i][k]表示第r行的状态为k,第r-1行的状态为i时候,前r行最多能够安装的大炮数量。

 

#include<iostream>#define max(a,b) ((a)>(b)?(a):(b))using namespace std;int map[105];// map[i]的二进制表示每一行的H分布状态。int cnt = 0, stk[65], sum[65];// cnt表示状态总数,stk[i]表示第i种状态,sum[i]表示第i种状态安装的大炮数量。int dp[105][65][65];bool ok(int x){        //  判断状态x是否符合,即是否会出现两个大炮间隔小于2。if(x & (x<<1)) return false;if(x & (x<<2)) return false;return true;}int getSum(int x){     //  求出状态x中安装了多少门大炮,x的二进制有几个1。int num = 0;while(x > 0){if(x & 1) num ++;x >>= 1;}return num;}void findStk(int n){   //  预处理求出多有可能的状态。for(int i = 0; i < (1<<n); i ++)if(ok(i)){stk[cnt] = i;sum[cnt ++] = getSum(i);}}int main(){int row, col, r, c, i, j, k;cin >> row >> col;memset(dp, -1, sizeof(dp));for(r = 0; r < row; r ++) {getchar();for(c = 0; c < col; c ++){char tmp;//cin >> tmp;scanf("%c", &tmp);if(tmp == 'H') map[r] |= (1<<c);}}findStk(col);for(i = 0; i < cnt; i ++)    //  第一行的状态特殊考虑。if(!(stk[i]& map[0]))dp[0][0][i] = sum[i];for(r = 1; r < row; r ++)for(i = 0; i < cnt; i ++){     //  枚举第r行的状态。if(stk[i]& map[r]) continue;for(j = 0; j < cnt; j ++){     //  枚举第r-1行的状态。if(stk[i]&stk[j]) continue;for(k = 0; k < cnt; k ++){     //  枚举第r-2行的状态。if(stk[i]&stk[k]) continue;if(dp[r-1][k][j] == -1) continue;dp[r][j][i] = max(dp[r][j][i], dp[r-1][k][j] + sum[i]);}}}int ans = 0;for(i = 0; i < cnt; i ++)for(j = 0; j < cnt; j ++)ans = max(ans, dp[row-1][i][j]);cout << ans << endl;return 0;}


 

原创粉丝点击