poj1185(状态压缩dp)

链接:点击打开链接

题意:司令部的将军们打算在N*M的网格地图上部署他们的炮兵部队。一个N*M的地图由N行M列组成,地图的每一格可能是山地(用"H" 表示),也可能是平原(用"P"表示),如下图。在每一格平原地形上最多可以布置一支炮兵部队(山地上不能够部署炮兵部队);一支炮兵部队在地图上的攻击范围如图中黑色区域所示:

poj1185(状态压缩dp)_第1张图片
如果在地图中的灰色所标识的平原上部署一支炮兵部队,则图中的黑色的网格表示它能够攻击到的区域:沿横向左右各两格,沿纵向上下各两格。图上其它白色网格均攻击不到。从图上可见炮兵的攻击范围不受地形的影响。
现在,将军们规划如何部署炮兵部队,在防止误伤的前提下(保证任何两支炮兵部队之间不能互相攻击,即任何一支炮兵部队都不在其他支炮兵部队的攻击范围内),在整个地图区域内最多能够摆放多少我军的炮兵部队。
代码:
#include <queue>
#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
char s[105];
int num[105],sum[105],str[105],dp[105][105][105];
int main(){                                     //dp[i][j][k]代表dii行状态是k,第i-1行状态是j
    int n,m,i,j,r1,r2,ans,cnt;
    while(scanf("%d%d",&n,&m)!=EOF){
        cnt=0;
        memset(dp,-1,sizeof(dp));
        memset(str,0,sizeof(str));
        for(i=0;i<(1<<m);i++){
            if((i&(i<<1))||(i&(i<<2)))
            continue;
            num[cnt]=i,sum[cnt++]=__builtin_popcount(i);
        }                                       //预处理出所有两个1中间没有0的状态,并记录每个状态中
        for(i=0;i<n;i++){                       //1的个数
            scanf("%s",s);
            for(j=0;j<m;j++)
            if(s[j]=='H')
            str[i]|=(1<<j);
        }                                       //将地形的状态处理出来
        for(i=0;i<cnt;i++)
        if(!(str[0]&num[i]))
        dp[0][0][i]=sum[i];                     //处理出第一行的种数
        for(i=1;i<n;i++){
			for(j=0;j<cnt;j++)
                if(!(str[i]&num[j])){
				for(r1=0;r1<cnt;r1++)
				if(!(num[j]&num[r1])){
					for(r2=0;r2<cnt;r2++)
					if(!(num[j]&num[r2])){      
						if(dp[i-1][r2][r1]!=-1)
						dp[i][r1][j]=max(dp[i][r1][j],dp[i-1][r2][r1]+sum[j]);
					}                           //dp[i][j][k]=max(dp[i][j][k],dp[i-1][k][l]+sum[j])
				}                               //l为i-2行的状态,sum为第i行的种数
			}
		}
        ans=0;
        for(i=0;i<cnt;i++)
        for(j=0;j<cnt;j++)
        ans=max(ans,dp[n-1][i][j]);
        printf("%d\n",ans);
    }
    return 0;
}

你可能感兴趣的:(源码,动态规划,ACM,poj)