骨牌摆放问题 POJ 2411(状态压缩DP)

题目:

给你\(n*m(1<=n,m<=11)\)的方格矩阵,要求用1*2的多米诺骨牌去填充,问有多少种填充方法。

比如下图是对于如下2x6的方格矩阵,可能的填充方案之一。

该如何使用动态规划的方式解决这道题呢?先了解一下状态压缩算法。

状态压缩通常是使用一个整数来表示一个集合,比如整数3,二进制表示为11,第一位状态为1,第二位状态为1,数字2的二进制表示为10,第一位的状态为1,第二位的状态为0,数字1的二进制表示为01,第一位为0,第二位为1,数字0的二进制可以表示为00,两位的状态都是0。

这就是说,状态可以保存在一个整数里面,对于状态压缩DP,其实也是用状态压缩后的整数表示一个维度,然后进行状态转移。

状态压缩DP:采用状态压缩算法的DP问题,也就是用整数表示集合,然后将该整数作为DP的一个维度来进行DP状态转移。

继续看题,其实这道题的解法并不简单,甚至于有些晦涩,光是理解状态压缩DP并不一定能够看懂源代码。

首先定义状态转移方程:

\(f(i, j) = f(i-1, k)\)之和,\(j, k\)属于\([0,1<<m)\)

且满足 \(j\&k==0\)

满足 $ j | k$是有连续个0的状态

f(i,j)表示第i行的状态为j时,前i行的方案总数。且定义状态1表示竖着的上面一部分,其他状态为0。

对于约束的理解是本题的关键,

约束1:j&k == 0 是因为不可能上下都是1。

约束2:j|k合并后状态0必须是连续偶数个,这个可以预先缓存起来,不用每次都算。详见代码注释。

源代码:

#include <limits.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <algorithm>
#include <bitset>
#include <iostream>
#include <set>
#include <string>
#include <vector>

using namespace std;

long long f[12][1 << 11];
int in_s[1 << 11];

// 注意这里一定要用64位的整数
long long solve(int n, int m) {
    for (int i = 0; i < 1 << m; ++i) {
        int cnt = 0, has_odd = 0;
        // 遍历m中的每一位
        for (int j = 0; j < m; ++j) {
            if (i >> j & 1) {
                //把cnt的值先存放到has_odd上,然后清零
                has_odd |= cnt, cnt = 0;
            } else {
                //如果是偶数个0,则肯定cnt最后为0,因为0^1=1 1^1=0
                cnt ^= 1;
            }
        }
        // 如果cnt=1则表明存在连续的奇数个0位
        in_s[i] = has_odd | cnt ? 0 : 1;
    }

    f[0][0] = 1;
    for (int i = 1; i <= n; ++i) {
        for (int j = 0; j < 1 << m; ++j) {
            f[i][j] = 0;
            for (int k = 0; k < 1 << m; ++k) {
                // 上面是1下面必须是0
                // 0必须是连续偶数个
                if ((j & k) == 0 && in_s[j | k]) {
                    // 满足要求后加上i-1行k的所有情况
                    f[i][j] += f[i - 1][k];
                }
            }
        }
    }

    // 最后一行的j所有位都是0
    return f[n][0];
}

int main() {
#ifdef __MSYS__
    freopen("test.txt", "r", stdin);
#endif

    int n, m;

    while (cin >> n >> m && n) {
        cout << solve(n, m) << endl;
    }
    return 0;
}
posted @ 2020-09-27 10:42  RioTian  阅读(347)  评论(0编辑  收藏  举报