[状压dp] 玉米田(状压dp)

0. 前言

相关：

• [状压dp] 蒙德里安的梦想(模板题+状压dp)

强相关：

• [状压dp] 小国王(状压dp)

1. 状压dp+棋盘式(基于连通性)

327. 玉米田

[状压dp] 小国王(状压dp) 是井字形，本题是十字形。

思路：

• 状态定义：
  • f[i][s]：摆放完前 i 行，且第 i 行摆放状态是 s 的方案数
• 状态计算：
  • 当 $s_i、s_{i-1}$ 二进制表示中无相邻 1 的时候，则横向无公共边
  • 当 $s_i\&s_{i-1}=0$ 时，则说明上下无公共边
  • 当 $s$ 二进制表示 1 的位置，土地状态恰好为 0，则该状态无效
  • 状态转移方程： f[i][s]+=f[i-1][si-1]
• 时间复杂度： $n\ast 2^n\ast 2^n=O(n{2}^{2n})=12\ast 2^{24}$ 看着妥妥超时，但是里面合法状态很少…依旧可以过

在此，head 数组存储对应关系，其实不必要存 state 数组的值。也可以存 state 数组中两个合法状态各自的下标。因为这也是一一映射的关系，且 dp 数组在枚举最后一维状态的时候，也可以存下标。这样代码能简洁很多…

代码：

#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>

using namespace std;

const int N = 14, M = 1 << 12, MOD = 1e8;

int n, m;
int g[N];               // 将地图每一行抽象为二进制表示
vector<int> state;
vector<int> head[M];
int f[N][M];

bool check(int x) {
    for (int i = 0; i + 1< m; ++i) 
        if ((x >> i & 1) && (x >> i + 1 & 1))
            return false;
    
    return true;
}

int main() {
    cin >> n >> m;
    
    for (int i = 1; i <= n; ++i) 
        for (int j = 0; j < m; ++j) {
            int t;
            cin >> t;
            g[i] += !t << j;            // 很巧妙,是1的地方不能选,方便做&运算
        }
    
    for (int i = 0; i < 1 << m; ++i)
        if (check(i))
            state.push_back(i);
    
    for (int i = 0; i < state.size(); ++i)
        for (int j = 0; j < state.size(); ++j) {
            int a = state[i], b = state[j];
            if ((a & b) == 0) 
                head[a].push_back(b);
        }
    
    f[0][0] = 1;
    for (int i = 1; i <= n + 1; ++i)
        for (int a = 0; a < state.size(); ++a)
            for (int b = 0; b < head[state[a]].size(); ++b) 
                if ((g[i] & state[a]) == 0)
                    f[i][state[a]] = (f[i][state[a]] + f[i - 1][head[state[a]][b]]) % MOD;
    
    cout << f[n + 1][0] << endl;
    return 0;
}

同理 [状压dp] 小国王(状压dp)，也可以用下标映射状态的思想，在 dp 数组中存下标即可。代码能简洁方便点。

存下标，简洁代码：

#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>

using namespace std;

const int N = 14, M = 1 << 12, MOD = 1e8;

int n, m;
int g[N];               // 将地图每一行抽象为二进制表示
vector<int> state;
vector<int> head[M];
int f[N][M];

bool check(int x) {
    for (int i = 0; i + 1< m; ++i) 
        if ((x >> i & 1) && (x >> i + 1 & 1))
            return false;
    
    return true;
}

int main() {
    cin >> n >> m;
    
    for (int i = 1; i <= n; ++i) 
        for (int j = 0; j < m; ++j) {
            int t;
            cin >> t;
            g[i] += !t << j;            // 很巧妙,是1的地方不能选,方便做&运算
        }
    
    for (int i = 0; i < 1 << m; ++i)
        if (check(i))
            state.push_back(i);
    
    for (int i = 0; i < state.size(); ++i)
        for (int j = 0; j < state.size(); ++j) {
            int a = state[i], b = state[j];
            if ((a & b) == 0) 
                head[i].push_back(j);               // 下标映射
        }
    
    f[0][0] = 1;
    for (int i = 1; i <= n + 1; ++i)
        for (int a = 0; a < state.size(); ++a)
            for (int b : head[a])
                if ((g[i] & state[a]) == 0)
                    f[i][a] = (f[i][a] + f[i - 1][b]) % MOD;
    
    cout << f[n + 1][0] << endl;
    return 0;
}