首页 > 代码库 > poj 3254 Corn Fields ,状态压缩DP

poj 3254 Corn Fields ,状态压缩DP

题目链接

题意:

一个矩阵里有很多格子,每个格子有两种状态,可以放牧和不可以放牧,可以放牧用1表示,否则用0表示,在这块牧场放牛,要求两个相邻的方格不能同时放牛,即牛与牛不能相邻。问有多少种放牛方案(一头牛都不放也是一种方案)



state[i] 表示对于一行,保证不相邻的方案


状态:dp[i][ state[j] ]  在状态为state[j]时,到第i行符合条件的可以放牛的方案数

状态转移:dp[i][ state[j] ] =Sigma dp[i-1][state‘] (state‘为符合条件的所有状态)


nclude<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

const int mod = 100000000;
int state[1<<13], cnt; //对于一行,保证不相邻的方案
int dp[20][1<<13];
int cur[20]; // cur[i]的值得二进制形式0表示能放,1表示不能放。

int main()
{
    int n, m, i, j, k;
    scanf("%d%d", &n, &m);
    int M = 1<<m;
    cnt = 0;
    for(i=0; i<M; ++i)
    {
        if(i&(i<<1)) continue; //二进制有相邻的1
        state[cnt++] = i;
    }

    for(i=0; i<n; ++i)
    {
        cur[i] = 0;
        int x;
        for(j=0; j<m; ++j)
        {
            scanf("%d", &x);
            if(!x) cur[i] |= 1<<j;
        }
    }

    memset(dp, 0, sizeof dp );
    for(i=0; i<cnt; ++i) if(!(cur[0]&state[i])) dp[0][state[i]] = 1;

    for(i=1; i<n; ++i)
    {
        for(j=0; j<cnt; ++j) if( dp[i-1][state[j]] ){
            for(k=0; k<cnt; ++k) if(!(state[k]&state[j]) && !(cur[i]&state[k]))
            {
                dp[i][state[k]] = (dp[i][state[k]] + dp[i-1][state[j]]) % mod;
            }
        }
    }
    int ans = 0;
    for(i=0; i<cnt; ++i)
        ans = (ans + dp[n-1][state[i]]) % mod;
    printf("%d\n", ans);
    return 0;
}