UESTC 882 冬马党 --状压DP
定义:dp[i][j]为状态为j时,第i行符合条件的状态数
转移方程:dp[i][j] += dp[i-1][t] //t为上一行状态,与当前行不冲突。
从第一行开始向下枚举,每次枚举当前行的状态和上一行的状态,如果不相邻或者未被地雷占据并且两行的关系是合法的,则加上方法数。
最后res = SUM(dp[n][s]) (s=0~S,为最后一行的状态)
代码:
#include <iostream> #include <cstdio> #include <cstring> #include <cmath> #include <algorithm> #define Mod 1000000007 using namespace std; #define N 100000000 //dp[i][j]:状态为j时,第i行符合条件的状态数 int wei[14]; int a[2070]; int dp[14][2070]; int checkadj(int state) { if(state & (state<<1)) //相邻 return 0; return 1; } int checkplace(int state,int i) { if(state & wei[i]) //被地雷占据 return 0; return 1; } int main() { int n,m,i,j,k,state; scanf("%d%d",&n,&m); memset(wei,0,sizeof(wei)); memset(dp,0,sizeof(dp)); for(i=1;i<=n;i++) { for(j=1;j<=m;j++) { scanf("%d",&state); if(!state) wei[i] += (1<<(m-j)); //被地雷占据赋为1 } } int Fst = 1<<m; k = 0; for(state=0;state<Fst;state++) //第一行 { if(checkadj(state)) { a[k++] = state; if(checkplace(state,1)) dp[1][k-1] = 1; } } for(i=2;i<=n;i++) { for(j=0;j<k;j++) //枚举上一行 { if(!checkplace(a[j],i-1)) continue; for(state=0;state<k;state++) //枚举这一行 { if(!checkplace(a[state],i) || (a[j] & a[state])) continue; dp[i][state] = (dp[i][state] + dp[i-1][j])%N; } } } int res = 0; for(i=0;i<k;i++) res = (res + dp[n][i])%N; printf("%d\n",res); return 0; }
作者:whatbeg
出处1:http://whatbeg.com/
出处2:http://www.cnblogs.com/whatbeg/
本文版权归作者和博客园共有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文连接,否则保留追究法律责任的权利。
更多精彩文章抢先看?详见我的独立博客: whatbeg.com