状压dp Poj - 3254 Corn Fields

状压dp

题意

有一块 m 行 n 列的农场地形，要在 ‘1’ 上养牛，并且相邻上下左右不能同时养牛。

问：有多少种方案来养牛？

思路

先看第一行样例 1 1 1，肯定不能相邻养牛，所以有以下几种情况

0 0 0	0
0 0 1	1
0 1 0	2
1 0 0	4
1 0 1	5

当加上第二行的时候，又分为两种情况，每种情况与上一行结合判断是否可行

得出状态转移方程：dp [ i ] [ state[ j ] ] = dp [ i - 1 ] [ state [ k1 ] ] + dp [ i - 1 ] [ state[ k2] ] +……+dp [ i - 1 ] [ state[ kn ] ]

#include<stdio.h>
#include<string.h>
int n,m,top;
int mod=100000000;
int dp[20][600];
int state[600],num[110];
int cur[20];
/**dp[i][state[j]] 对于前 i 行，第 i 行采用第 j 种状态时
可以得到的方案总数
dp[i][state[j]]=dp[i-1][state[k1]]+dp[i-1][state[k2]]+……+dp[i-1][state[kn]]
*/
bool judge(int x)
{
    if(x&x<<1)return false;
    return true;
}

void init()
{
    top=0;
    int total=1<<n;
    for(int i=0; i<total; i++)
    {
        if(judge(i))state[++top]=i;/**存储可以用的状态*/
    }
}
bool fit(int x,int k)
{
    if(x&cur[k])return false;
    return true;
}
int main()
{
    scanf("%d%d",&m,&n);
    init();
    memset(dp,0,sizeof(dp));

       for(int i = 1; i <= m; ++i){
           cur[i] = 0;
           int num;
           for(int j = 1; j <= n; ++j){  //输入时就要按位来存储，cur[i]表示的是第i行整行的情况，每次改变该数字的二进制表示的一位
                scanf("%d",&num);  //表示第i行第j列的情况（0或1）
               if(num == 0) //若该格为0
				   cur[i] +=(1<<(n-j)); //则将该位置为1（注意要以相反方式存储，即1表示不可放牧
           }
       }
       
       for(int i = 1;i <= top;i++){
           if(fit(state[i],1)){  //判断所有可能状态与第一行的实际状态的逆是否有重合
                dp[1][i] = 1;  //若第1行的状态与第i种可行状态吻合，则dp[1][i]记为1
           }
       }
    for(int i=2; i<=m; i++)
    {
        for(int k=1; k<=top; k++)
        {
            if(!fit(state[k],i))
                continue;/**第 i 行是 K 状态*/
            for(int j=1; j<=top; j++)
            {
                if(!fit(state[j],i-1))continue;/**i-1 行是所有可行状态 j */
                if(state[k]&state[j])continue;/**两种状态不冲突*/
                dp[i][k]=(dp[i][k]+dp[i-1][j])%mod;/**加上 i-1 行所有的状态*/
            }
        }
    }
    int ans=0;
    for(int i=1;i<=top;i++)
    {
        ans=(ans+dp[m][i])%mod;
    }
    printf("%d\n",ans);
    return 0;
}

附上学习博客：https://blog.csdn.net/harrypoirot/article/details/23163485