洛谷 P1879 [USACO06NOV] Corn Fields G 题解 状压dp

[USACO06NOV] Corn Fields G

题目描述

农场主 $\mathrm{John}$ 新买了一块长方形的新牧场，这块牧场被划分成 $M$ 行 $N$ 列 $(1\le M\le 12,1\le N\le 12)$，每一格都是一块正方形的土地。 $\mathrm{John}$ 打算在牧场上的某几格里种上美味的草，供他的奶牛们享用。

遗憾的是，有些土地相当贫瘠，不能用来种草。并且，奶牛们喜欢独占一块草地的感觉，于是 $\mathrm{John}$ 不会选择两块相邻的土地，也就是说，没有哪两块草地有公共边。

$\mathrm{John}$ 想知道，如果不考虑草地的总块数，那么，一共有多少种种植方案可供他选择？（当然，把新牧场完全荒废也是一种方案）

输入格式

第一行：两个整数 $M$ 和 $N$，用空格隔开。

第 $2$ 到第 $M+1$ 行：每行包含 $N$ 个用空格隔开的整数，描述了每块土地的状态。第 $i+1$ 行描述了第 $i$ 行的土地，所有整数均为 $0$ 或 $1$ ，是 $1$ 的话，表示这块土地足够肥沃，$0$ 则表示这块土地不适合种草。

输出格式

一个整数，即牧场分配总方案数除以 $100,000,000$ 的余数。

样例 #1

样例输入 #1

2 3
1 1 1
0 1 0

样例输出 #1

9

题目链接

洛谷P1879——传送门

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

ll m, n;          // m行n列
vector<ll> s[16]; // 行合法的二进制压缩的状态(状压s)
ll dp[16][5006];  // dp[i][j]，表示第i行，行状态为j
ll mp[16][16];    // 记录土地的贫瘠或肥沃
const ll mod = 1e8;

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    cin >> m >> n;
    for (int i = 1; i <= m; i++)
    {
        for (int j = 0; j < n; j++)
        {
            cin >> mp[i][j];
        }
    }

    for (ll i = 0; i < (1 << n); i++) // 枚举二进制状态，从而选取符合条件的状态加入数组s
    {
        for (ll j = 1; j <= m; j++)
        {
            bool ok = 1;
            for (ll k = 0; k < n; k++)
            {
                if (((i >> k) & 1) == 1 && mp[j][k] == 0) // 如果在贫瘠的土地上种草，则不符合条件
                {
                    ok = 0;
                    break;
                }
            }
            if ((((i >> 1) | (i << 1)) & i) == 0 && ok == 1) // 如果没有相邻（即 左、右 两种相邻关系）的国王，表示该状态在一行中是符合条件的，同时要满足种草的土地是肥沃的
            {
                s[j].push_back(i); // 第j行符合条件的状态
            }
        }
    }

    for (ll j = 0; j < s[1].size(); j++)
    {
        dp[1][s[1][j]] = 1; // 初始化第一行每种状态的方案数为1
    }

    for (ll i = 2; i <= m; i++) // 枚举行
    {
        for (ll k1 = 0; k1 < s[i].size(); k1++) // 枚举当前行合法的行状态
        {
            ll s1 = s[i][k1];
            for (ll k2 = 0; k2 < s[i - 1].size(); k2++) // 枚举上一行合法的行状态
            {
                ll s2 = s[i - 1][k2];
                if ((s2 & s1) == 0) // 上下不相邻则符合条件
                {
                    dp[i][s1] += dp[i - 1][s2];
                    dp[i][s1] %= mod;
                }
            }
        }
    }

    ll ans = 0;
    for (ll i = 0; i < s[m].size(); i++)
    {
        ans += dp[m][s[m][i]]; // 统计枚举到最后一行后的符合条件的总方案数量
        ans %= mod;
    }

    cout << ans << '\n';

    return 0;
}