YbtOJ 状压DP问题课堂过关 例1 种植方案【状压DP】

题目

思路

本题是状压DP模板题
枚举相邻的两行状态就好了
注意要先把所有合法的状态筛出来。

代码

#include
#include
using namespace std;
int q[1<<12],v[30],tot,f[30][1<<12];
int n,m,a[30][30],ans;
int main()
{
	cin>>n>>m;
	for(int i=0; i<=(1<<m)-1; i++)
	 if(((i<<1)&i)==0)
	   q[++tot]=i;
	for(int i=1; i<=n; i++)
	 for(int j=1; j<=m; j++)
	  {
	    scanf("%d",&a[i][j]);
	    v[i]=(v[i]<<1)+a[i][j];
	  }
	f[0][0]=1;
	for(int i=1; i<=n; i++)
	 {
	 	for(int j=1; j<=tot; j++)
	 	 if((v[i]&q[j])==q[j])
	 	  {
	 	  	for(int k=1; k<=tot; k++)
	 	  	 {
	 	  	 	if((q[j]&q[k])==0)
	 	  	 	 {
	 	  	 	 	f[i][q[j]]+=f[i-1][q[k]];
	 	  	 	 	f[i][q[j]]%=100000000;
				 }
			 }
		  }
	 }
	for(int i=1; i<=tot; i++)
	   ans=(ans+f[n][q[i]])%100000000;
	printf("%d",ans);
	return 0;
}