首页 > 代码库 > POJ 2411 Mondriaan's Dream (dp + 状态压缩)

POJ 2411 Mondriaan's Dream (dp + 状态压缩)

链接:http://poj.org/problem?id=2411

题意:题目描述:用1*2 的矩形通过组合拼成大矩形,求拼成指定的大矩形有几种拼法。

参考博客:http://blog.csdn.net/shiwei408/article/details/8821853

思路:我看了上面的博客,想了很久才明白是如何处理状态的。

由于是1 * 2,所以可以通过相邻两行的转化关系来推导。

两行铺不铺砖可以用二进制来表示,但是如果暴力枚举,大概有2^10 * 2 ^ 10 次那么多状态(虽然其中有很多状态是没有用的)。

所以采用dfs来枚举各种可行状态:

状态标记 横放和竖放的下一个均为1,竖放的上一个和不放置为0 ,每行可以转化为1个2进制数。当这一行访问结束时,就会得到上一行状态,和该行状态,因为所有情况都是我们设置的,所以pre状态一定会转化为now状态

对于每一个位置,我们有三种放置方法:1. 竖直放置2. 水平放置3. 不放置

d为当前列号 ,初始化d, now, pre都为0。now为当前行,pre为当前行的上一行

1. d = d + 1, now << 1 | 1, pre << 1;   // 竖直放置,当前行该列为1,上一行该列置为0

2. d = d + 2, now << 2 | 3, pre<< 2 | 3;  // 横放 都为11(因为当两行的状态往下推时,必须要保证pre这一行摆满,所以pre也都为11)

3. d = d + 1, now << 1, pre<< 1 | 1;    // 上一行该列置为1,不能竖放,不放置的状态

因为转移状态有很多种,所以用dfs去枚举各种可行的状态。

最后在转移的时候,dp[0][(1 << w) - 1] = 1 表示只有第0行全部铺满才行。只需要算到 h - 1 行,然后 h 行全部铺满为答案。

代码:

/*
ID: wuqi9395@126.com
PROG:
LANG: C++
*/
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<cmath>
#include<cstdio>
#include<vector>
#include<string>
#include<fstream>
#include<cstring>
#include<ctype.h>
#include<iostream>
#include<algorithm>
#define INF (1<<30)
#define PI acos(-1.0)
#define mem(a, b) memset(a, b, sizeof(a))
#define rep(i, n) for (int i = 0; i < n; i++)
#define debug puts("===============")
#define eps (1e-6)
typedef long long ll;
using namespace std;
const int maxn = 13;
int w, h, cnt;
ll dp[13][2100];
int sta[14000][2];
void dfs(int l, int now, int pre) {
    if (l > w) return ;
    if (l == w) {
        sta[cnt][0] = pre, sta[cnt++][1] = now;
        return ;
    }
    dfs(l + 2, (now << 2) | 3, (pre << 2) | 3);
    dfs(l + 1, (now << 1) | 1, pre << 1);
    dfs(l + 1, now << 1, (pre << 1) | 1);
}
int main () {
    while(~scanf("%d%d", &h, &w), h || w) {
        if (h < w) swap(h, w);
        cnt = 0;
        dfs(0, 0, 0);
        memset(dp, 0, sizeof(dp));
        dp[0][(1 << w) - 1] = 1;
        rep(i, h) {
            rep(j, cnt) {
                dp[i + 1][sta[j][1]] += dp[i][sta[j][0]];
            }
        }
        printf("%lld\n", dp[h][(1 << w) - 1]);
    }
    return 0;
}