炮兵阵地 POJ - 1185(状压DP)
G - 炮兵阵地
题目链接:POJ - 1185POJ-3254的升级版;
题意:N*M的地图标有H,G, H不能建炮兵阵地, P可以,炮兵阵地之间至少要间距两个格子(POJ-3254是不能相邻);若用1表示有炮兵阵地, 0表示无, 则1001001可行, 1100和1010不行;
思路:POJ-3254的每行只被上一行影响, 这道题的每行被前两行影响;
#include <iostream> #include <algorithm> #include <string.h> using namespace std; int n, m, top, dp[105][100][100], fild[105], state[100], cnt[100];//打表可知, 符合条件的状态只有60多个; bool ok(int x){ if(x&(x<<1)||x&(x<<2)) return false; return true; } int Count(int x){ int ans=0; while(x){ if(x&1) ans++; x>>=1; } return ans; } bool fit(int x, int i){ if(x&fild[i]) return false; return true; } void init(){ top=0; for(int i=0; i<(1<<m); i++){ if(ok(i)){ state[top]=i; cnt[top++]=Count(i); } } } int main(){ char ch; cin >>n >> m; init(); for(int i=1; i<=n; i++){ fild[i]=0; for(int j=1; j<=m; j++){ cin >>ch; if(ch=='H') fild[i]+=(1<<(m-j)); } } memset(dp, 0, sizeof(dp)); for(int i=0; i<top; i++){ if(fit(state[i], 1)) dp[1][i][0]=cnt[i]; } for(int i=0; i<top; i++){ if(!fit(state[i], 2)) continue; for(int j=0; j<top; j++){ if(!fit(state[j], 1)) continue; if(state[i]&state[j]) continue; dp[2][i][j]=max(dp[2][i][j], dp[1][j][0]+cnt[i]); } } for(int i=3; i<=n; i++){ for(int j=0; j<top; j++){ if(!fit(state[j], i)) continue; for(int k=0; k<top; k++){ if(!fit(state[k], i-1)) continue; if(state[j]&state[k]) continue; for(int l=0; l<top; l++){ if(!fit(state[l], i-2)) continue; if(state[j]&state[l]) continue; if(state[k]&state[l]) continue; dp[i][j][k]=max(dp[i][j][k], dp[i-1][k][l]+cnt[j]); } } } } int ans=0; for(int i=0; i<top; i++){ for(int j=0; j<top; j++){ ans=max(ans, dp[n][i][j]); } } cout <<ans <<endl; return 0; }