PS:本道题目建议在对状压dp有一定了解的基础下学习,如有不懂可以先去学一下状压dp入门
题目大意:给你n*m个格子,有些格子可以用来部署军队,用P表示,有些则不能,用H表示,如果在一个格子上部署了军队,则上下左右各2个格子都不能部署军队,也就是呈十字架状,看到数据范围(n<=100,m<=10)很容易想到使用状压dp,因为m列数最大只有10,我们可以压缩每一行的状态,最大只有(1<<10)-1种状态,但是由于这一行的状态对下一行以及下两行都有影响,我们需要一个三维数组来保存状态,dp[i][j][k],代表第i行状态为j,上一行状态为k最大能部署的军队数,那么我们应该就能推出状态转移方程:
dp[i][j][k]=max(dp[i][j][k],dp[i-1][k][l]+num[i])(l为上两行的状态,num[i]为状态为i时能部署的军队数)
但是有一个问题,就是n最大为10,但是我们的数组好像还是开不下,dp[110][1<<10][1<10]显然是超内存的,怎么办呢?我们其实可以抓住如果在一个格子上部署军队则左右两边各2个格子都不能部署军队这一点来预处理进而达到缩小内存的方法,那么怎么预处理呢,其实和那道入门题的判断方法是一样的,我们将某一个状态分别右移1位,2位,这样便得到三个状态,然后两两相与,如果都等于0说明这个状态是合法的,那么这样下来最多的状态数也只有60个,这样数组就足够用了
具体实现方法如下
1 void init() 2 { 3 for(int i=0;i<(1<<m);i++) 4 { 5 if(!(i&(i>>1)) && !(i&(i>>2)) && !((i>>1)&(i>>2))) 6 { 7 can[++cnt]=i; 8 num[cnt]=get(i); 9 } 10 } 11 //cout<<cnt<<endl;-------算出来最大只有60 12 }
然后就是保存每一行是否可以部署军队的操作,这里记得要将P和H反过来存,若为正着存,则后面不部署军队的情况是考虑不到的,应该很好理解吧
in(n);in(m); for(int i=1;i<=n;i++) { for(int j=1;j<=m;j++) { char ch; cin>>ch; if(ch=='H') cur[i]|=1<<(j-1);//反过来存状态 } }
然后就是dp部分了,因为从第三行开始,后面的行的状态都与前面一行及两行的状态有关,因此我们先处理出第一行以及第二行的状态,以便后面的递推
第一行的时候我们是不需要判断前面的行数的,因此只需要判断这一行的状态是否和合法就行了
而第二行的时候我们则需要根据在第二行状态合法的前提下还需要根据第一行的状态看有没有冲突的的情况,这种情况是要舍去的,其他的就没什么好讲的了
for(int i=1;i<=cnt;i++) { if(!(cur[1]&can[i])) { dp[1][i][0]=max(dp[1][i][0],num[i]); tot=max(tot,dp[1][i][0]); } } //第一行的判断 for(int i=1;i<=cnt;i++) { if(!(cur[2]&can[i])) { for(int j=1;j<=cnt;j++) { if(!(cur[1]&can[j]) && !(can[i]&can[j])) { dp[2][i][j]=max(dp[2][i][j],dp[1][j][0]+num[i]); tot=max(tot,dp[2][i][j]); } } } } //第二行的判断
其实后面的行数跟前面的判断差不多,就是先判断本行再判断上一行,再判断上两行,最判断这三个状态有没有冲突,只要有一个存在冲突,这种状态就是不合法的,那么我们就把它舍去
最后上个完整版代码
#include<iostream> #include<cstdio> #include<cmath> #include<string> #include<cstring> #define in(i) (i=read()) using namespace std; int read() { int ans=0,f=1; char i=getchar(); while(i<'0'||i>'9') { if(i=='-') f=-1; i=getchar(); } while(i>='0'&&i<='9') { ans=(ans<<1)+(ans<<3)+i-'0'; i=getchar(); } return ans*f; } int dp[110][65][65]; int can[65]; int cur[110]; int num[65]; int n,m,cnt,tot; int get(int x) { int ans=0; for(int i=1;i<=m;i++) { if((x&(1<<(i-1)))!=0) ans++;//判断是否可以部署军队,若可以则ans++ } return ans; } void init() { for(int i=0;i<(1<<m);i++) { if(!(i&(i>>1)) && !(i&(i>>2)) && !((i>>1)&(i>>2)))//相邻之间没有1 { can[++cnt]=i; num[cnt]=get(i); } } //cout<<cnt<<endl;-------算出来最大只有60 } int main() { in(n);in(m); for(int i=1;i<=n;i++) { for(int j=1;j<=m;j++) { char ch; cin>>ch; if(ch=='H') cur[i]|=1<<(j-1);//记得反过来存 } } init();//预处理 for(int i=1;i<=cnt;i++)//不用判断上一行 { if(!(cur[1]&can[i])) { dp[1][i][0]=max(dp[1][i][0],num[i]); tot=max(tot,dp[1][i][0]); } } //处理第一行
for(int i=1;i<=cnt;i++)//只用判断上一行 { if(!(cur[2]&can[i])) { for(int j=1;j<=cnt;j++) { if(!(cur[1]&can[j]) && !(can[i]&can[j])) { dp[2][i][j]=max(dp[2][i][j],dp[1][j][0]+num[i]); tot=max(tot,dp[2][i][j]); } } } } //处理第二行
for(int i=3;i<=n;i++)//枚举行数,对后n-2行进行处理 { for(int j=1;j<=cnt;j++)//枚举第i行的状态 { if(!(cur[i]&can[j]))//判断本行是否有冲突 { for(int k=1;k<=cnt;k++)//若无则枚举下一行的状态 { if(!(cur[i-1]&can[k]))//判断上一行状态是否有冲突 { for(int l=1;l<=cnt;l++)//枚举上上一行状态 { if(!(cur[i-2]&can[l])) { if(!(can[j]&can[k]) && !(can[j]&can[l]) && !(can[k]&can[l]))//若都无冲突 { dp[i][j][k]=max(dp[i][j][k],dp[i-1][k][l]+num[j]);//状态转移 tot=max(tot,dp[i][j][k]); } } } } } } } } cout<<tot<<endl; return 0; }