题意:农夫有一块地,被划分为m行n列大小相等的格子,其中一些格子是可以种植作物的(用1标记),其他格子则不能种植(用0标记),并且要求相邻格子不能同时都有种植作物。现在输入数据给出这块地的大小及可否种植的情况,求该农夫有多少种种植方案可以选择(注意:任何格子都不种植也是一种选择,不要忘记考虑!)
状压DP入门题,有个博客讲得很详细:
https://www.cnblogs.com/real-l/p/8562834.html
状压DP的含义:
在我们解决动态规划题目的时候,dp数组最重要的一维就是保存状态信息,但是有些题目它虽具有dp的特性,但状态较多,如果直接保存的可能需要三维甚至多维数组,这样在题目允许的内存下势必是开不下的,那么我们能不能想个办法,把它压缩成一维呢?对,二进制.一般的动规题目数据范围都不会太大,那么就可以把几个状态全部压缩成一个二进制数保存下来,这样就大大节省了空间,来允许我们进行其他的操作,这就叫做状态压缩.运用状态压缩来保存状态的dp就叫做状压dp,这类dp一般数据范围有一项很小(好像是不超过16吧),看到这种数据范围就可以往状压上想
其实状压dp就是一种枚举,是最暴力的一种dp.
引自上述博客。
本题的详解可看上述博客,在此仅罗列一些要点。
本题关键是如何枚举。
1、设置cur[i],表示第i行的状态,用一个十进制数表示;
2、设置can[j],表示在不考虑行的状态,以及各行之间的影响的前提下,可以种植作物的状态。can[j]保存的是一个十进制数,它拆分成二进制就是一个方案,如can[j]=5=101,就表示列数等于3时,可种植的一个方案,1表示种了,0表示没种。
那么怎么把can[j]求出来呢?只需两行代码就可以了
int cnt=0,tot=1<<n;
for(int i=0;i<tot;i++) if(!(i&(i<<1))) can[++cnt]=i;
从0开始枚举,将i拆分成二进制(用位运算可实现),如果(i&(i<<1))==0,说明此时i的二进制形式没有相邻的两个1,因此是满足要求的方案,记录下i。
3、注意我们can[j]记录的是不考虑行的状态以及各行之间的影响的时候的方案数,下面我们就要引入行的状态了。假设第i行输入是101,那么我们0,1互换,成010→2,cur[i]=2,这里解释一下为什么要互换。因为0是不能种的,1是可以的,那么0->1我们可以理解为原本0的位置不能种的原因是因为这里已经种作物了,而1->0则是1的位置还没有种作物,所以才可以种。那么接下来只要cur[i]&can[j]==0,即cur[i]与can[j]的二进制表示中,相同的位置上没有同时出现1,就说明can[j]满足i行的状态了。
4、接下来我们再引入各行之间的影响,根据题意,种了作物的位置相邻的位置就不能再种,因此,对于某一行,合法的方案就是该方案既满足该行的状态,又满足它的二进制表示中,与相邻行的方案的二进制表示,不会在相同位置上同时出现1。
代码如下:
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long ll;
const int mod=1e8;
const int maxn=1e3+5;
ll dp[13][4500]; //dp[i][j]表示第i行的方案为j时,总共用有多少种合法的方案
int cur[13],can[4500];
int main()
{
int m,n;
cin>>m>>n;
for(int i=1;i<=m;i++)
{
int base=1,temp=0;
for(int j=1;j<n;j++) base*=2;
for(int j=1;j<=n;j++)
{
int a;
cin>>a;
if(a) a=0;
else a=1;
temp+=a*base;
base/=2;
}
cur[i]=temp;
}
int cnt=0,tot=1<<n;
for(int i=0;i<tot;i++) if(!(i&(i<<1))) can[++cnt]=i;
memset(dp,0,sizeof(dp));
for(int i=1;i<=cnt;i++) if(!(cur[1]&can[i])) dp[1][can[i]]=1; //我们只需初始化第一行,接下来就可以根据第一行,推出每一行了
for(int i=2;i<=m;i++)
{
for(int j=1;j<=cnt;j++)
if(!(cur[i-1]&can[j])) //上一行的方案合法
{
for(int k=1;k<=cnt;k++)
if(!(cur[i]&can[k])&&!(can[j]&can[k])) //当前行的方案也合法
dp[i][can[k]]=(dp[i][can[k]]+dp[i-1][can[j]])%mod;
}
}
ll sum=0;
for(int i=1;i<=cnt;i++)
sum=(sum+dp[m][can[i]])%mod;
cout<<sum<<endl;
return 0;
}