poj 3254 Corn Fields(状压DP+滚动数组)

传送门:http://poj.org/problem?id=3254

这题是状压DP的入门题吧。

用了滚动数组优化了一下,0msAC。

具体解释直接看代码:

<pre name="code" class="cpp">#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

const int INF=0x3f3f3f3f;
const int mod=100000000;
const int maxn=15;
int T,n,m;
int state[1<<maxn],map[maxn];
int dp[2][1<<maxn];
int stateNum;

inline bool ok1(int x){//判断x状态是否可行
	return !(x&(x<<1));
}

inline bool ok2(int x,int y){//判断x状态与第y行状态是否可行
	return !(x&map[y]);
}

void init(){
	memset(dp,0,sizeof(dp));
	memset(map,0,sizeof(map));
	stateNum=0;
	int cnt=(1<<m);
	for(int i=0;i<cnt;i++){//筛除具有相邻的1的状态
		if(ok1(i))
			state[++stateNum]=i;
	}
}

int main(){
#ifndef ONLINE_JUDGE
	freopen("test.in","r",stdin);
	freopen("test.out","w",stdout);
#endif
	while(~scanf("%d%d",&n,&m)){
		init();
		int x;
		for(int i=1;i<=n;i++){
			for(int j=1;j<=m;j++){
				scanf("%d",&x);
				if(!x) map[i]|=(1<<(j-1));
			}
		}
		for(int i=1;i<=stateNum;i++){//初始化,如果状态i和第一行状态不冲突,则置dp[1][i]=1
			if(ok2(state[i],1))
				dp[1][i]=1;
		}
		for(int i=2;i<=n;i++){
			for(int j=1;j<=stateNum;j++){
				if(!ok2(state[j],i-1)) continue;//判断状态j与第i-1行是否冲突
				for(int k=1;k<=stateNum;k++){
					if(!ok2(state[k],i)) continue;//判断状态k与第i行是否冲突
					if(state[j]&state[k]) continue;//判断状态j与状态k是否冲突
					dp[i%2][k]=(dp[i%2][k]+dp[(i-1)%2][j])%mod;
				}
				dp[(i-1)%2][j]=0;//用滚动数组的时候要清零,否则会影响下次更新
			}
		}
		int ans=0;
		for(int i=1;i<=stateNum;i++){
			ans=(ans+dp[n%2][i])%mod;
		}
		printf("%d\n",ans);
	}
	return 0;
}


 
 

你可能感兴趣的:(poj 3254 Corn Fields(状压DP+滚动数组))