状态压缩DP —— 蒙德里安的梦想

蒙德里安的梦想

学习自Acwing，附Acwing原题

题目：

求把 N×M 的棋盘分割成若干个 1×2 的长方形，有多少种方案。

例如当 N=2，M=4 时，共有 5 种方案。当 N=2，M=3 时，共有 3 种方案。

输入格式
输入包含多组测试用例。

每组测试用例占一行，包含两个整数 N 和 M。

当输入用例 N=0，M=0 时，表示输入终止，且该用例无需处理。

输出格式
每个测试用例输出一个结果，每个结果占一行。

数据范围
1≤N,M≤11
输入样例：

1 2
1 3
1 4
2 2
2 3
2 4
2 11
4 11
0 0

输出样例：

1
0
1
2
3
5
144
51205

思路分析：

代码：

#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int N = 1 << 11;  //一列中最多存在的方案数是2^11，因为n最大等于11
int n, m;
long long f[55][N]; 
int st[N]; //判断对于j这种方法是否能放竖的
int main()
{

    while (cin >> n >> m && n != 0)
    {
        //对于一列的方案有2^n种,判断是否能放下竖的
        //判断方法只要连续空着的位置是偶数即可。
        for (int i = 0; i < 1 << n; i++)
        {
            int cnt = 0;
            st[i] = 1;
            for (int j = 0; j < n; j++)
            {
                if (i >> j & 1 == 1)
                {
                    if (cnt % 2 == 1)st[i] = 0;
             //       cnt = 0;
                }
                else cnt++;
            }
            if (cnt % 2 == 1)st[i] = 0;
        }


        memset(f, 0, sizeof(f));
        f[0][0] = 1;    //初始化，对于0 - 1列放法只有1种，那就是一个也不放，因为不存在从第0 - 1列到0列的木块
        for (int i = 1; i <= m; i++)
        {
            for (int j = 0; j < 1 << n; j++)
            {
                for (int k = 0; k < 1 << n; k++)
                {
                    //从第i-2列到第i - 1列和第i - 1列到第i列不会存在冲突
                    //第i - 1 列可以放下竖的木块
                    if (st[j | k] && (j & k) == 0)  
                    {
                        f[i][j] += f[i - 1][k];
                    }
                }
            }
        }
        cout << f[m][0] << '\n';    //前m - 1 已经放好，且伸到第m列的状态是0（即数量为0），即么就说明我们放满了
    }
    return 0;
}

我们可以把判断是否合理预先处理一下，之后就直接转移即可，不用在进行判断，这样可以进一步优化。
贴一下y总的优化代码：

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

using namespace std;

typedef long long LL;

const int N = 12, M = 1 << N;

int n, m;
LL f[N][M];
vector<int> state[M];
bool st[M];

int main()
{
    while (cin >> n >> m, n || m)
    {
        for (int i = 0; i < 1 << n; i ++ )
        {
            int cnt = 0;
            bool is_valid = true;
            for (int j = 0; j < n; j ++ )
                if (i >> j & 1)
                {
                    if (cnt & 1)
                    {
                        is_valid = false;
                        break;
                    }
                    cnt = 0;
                }
                else cnt ++ ;
            if (cnt & 1) is_valid = false;
            st[i] = is_valid;
        }

        for (int i = 0; i < 1 << n; i ++ )
        {
            state[i].clear();
            for (int j = 0; j < 1 << n; j ++ )
                if ((i & j) == 0 && st[i | j])
                    state[i].push_back(j);
        }

        memset(f, 0, sizeof f);
        f[0][0] = 1;
        for (int i = 1; i <= m; i ++ )
            for (int j = 0; j < 1 << n; j ++ )
                for (auto k : state[j])
                    f[i][j] += f[i - 1][k];

        cout << f[m][0] << endl;
    }

    return 0;
}

作者：yxc
链接：https://www.acwing.com/activity/content/code/content/64200/
来源：AcWing
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。