hihocoder 状态压缩二

来源:互联网 发布:淘宝代运营服务商 编辑:程序博客网 时间:2024/05/21 17:34

注释都在代码里。挺经典状态压缩,将二维转化成一维。结合dfs生成状态。

#include <iostream>#include <cstdio>using namespace std;const long long MOD=1000000007;long long dp[1010][64]={0};int n,m,h;void dfs(int base,int sta,int k,int i){if(k>=m){if(k==m){if(base&(h>>(k)))sta<<=1;elsesta=(sta<<1)+1;}dp[i][sta]=(dp[i][sta]+dp[i-1][base>>1])%MOD;//将上一行的合法状态中的种类数加进来return ;}if(base&(h>>k)) dfs(base,sta<<1,k+1,i);else{dfs(base,sta<<1|1,k+1,i);if((base&(h>>k+1))==0){if(k==(m-1)) dfs(base,sta<<2,k+2,i);else dfs(base,sta<<2,k+2,i);}}}//搜索上一行中可以产生base即j的状态,为便于处理最后一列,base=j<<1|1int main(){scanf("%d%d",&n,&m);int end=1<<m;//表示一行最多有多少种状态h=1<<m+1;if(m==3)dp[1][1]=dp[1][4]=dp[1][7]=1;else if(m==4)dp[1][0]=dp[1][12]=dp[1][3]=dp[1][15]=dp[1][9]=1;else{dp[1][1]=dp[1][4]=dp[1][16]=dp[1][7]=1;dp[1][19]=dp[1][25]=dp[1][28]=dp[1][31]=1;}//根据m不同对第一行进行不同的初始化for(int i=2;i<=n;i++){//从第二行开始,由上一行生成for(int j=0;j<end;j++){if(!dp[i-1][j]) continue;dfs(j<<1|1,0,1,i);//j表示第i行的状态}}printf("%lld\n",dp[n][0]);//对于状态的设定,可以影响到下一行同一列的设为1,否则为0return 0;}


 

0 0
原创粉丝点击