ACWing基础课
蒙德里安的梦想
求把 N×MN×M 的棋盘分割成若干个 1×21×2 的的长方形,有多少种方案。
例如当 N=2,M=4N=2,M=4 时,共有 55 种方案。当 N=2,M=3N=2,M=3 时,共有 33 种方案。
如下图所示:
输入格式
输入包含多组测试用例。
每组测试用例占一行,包含两个整数 NN 和 MM。
当输入用例 N=0,M=0N=0,M=0 时,表示输入终止,且该用例无需处理。
输出格式
每个测试用例输出一个结果,每个结果占一行。
数据范围
1≤N,M≤111≤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
原题链接:<291. 蒙德里安的梦想 - AcWing题库>
分析
典型的状态压缩
-
首先明确所有的方案数等于放1×2的长方形的方案数,因为当所有的横放的长方形放好后,竖放的长方形就固定了
-
状态表示:用 f [ i ] [ j ] f[i][j] f[i][j]表示状态,具体含义:当第 i − 1 i-1 i−1列摆好后,所有从第 i − 1 i-1 i−1列伸到第 i i i列的状态。若伸出为1,未伸出为0, j j j为状态的十进制表示,如:
-
状态计算:第 i 列固定为 j j j,第 i − 2 i-2 i−2 列转移到第$ i-1 列 的 : 列的: 列的:f[i-1][k] , 当 ,当 ,当f[i-1][k]的状态合法时,可以直接转移
闫氏DP分析:
-
判断状态是否合法
-
同一行中小方块不会有重叠:当k|j==0时表示两种状态的每一位都想法,两列的每一行都没有重叠,如下图所示
-
一列中空余小方格数量为偶数,留给竖型小方格填充,可以先预处理出来
判断条件:(j&k )==0 && st[ j| k] )
为什么是j&k==0以及 j| k?这里参考大佬的解释:
https://www.acwing.com/solution/content/28088/
j&k:当j&k==0时,说明两种状态的每一位都不同时为1,说明第i-2和第i-1列任意一行不同时为1。
j|k:我们要考虑的是第i-1列(第i-1列是这里的主体)中从第i-2列横插过来的,还要考虑自己这一列。比如第i-2列插过来的是k=10101,第i-1列插出去到第i列的是 j =01000,那么合在第i-1列,到底有多少个1呢?自然想到的就是这两个操作共同的结果:两个状态或。 j | k = 01000 | 10101 = 11101
-
-
最后状态 f [ m ] [ 0 ] f[m][0] f[m][0]
前 0 0 0~ m − 1 m-1 m−1列都已经摆满,且第 m m m列无小方块伸出
AC code
//蒙德里安的梦想
#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
typedef long long LL;
const int N=12,M=1<<N;
int n,m;
bool st[M]; //st表示当前状态是否合法:当某一列横向长方形摆放完成时
//该列连续未摆的连续空格是否为偶数(这些空格只能用竖的长方形填充)
LL f[N][M]; //f[i][j]表示当第i-1列摆好时,伸出到第i列的状态(十进制存储二进制)
vector<int> state[M]; //state表示当前状态下,所有可以转移到状态M的合法状态
int main()
{
while(cin>>n>>m,n||m) //n||m表示n、m不同时为0时继续读入
{
for(int i=0;i<1<<n;i++) //该层循环统计每个状态是否合法
{
int cnt=0;
bool is_valid=true;
for(int j=0;j<n;j++) //循环遍历i的每一位
{
if(i>>j&1) //i的第j位&1结果为1,表示 i的第j位为1
{
if(cnt&1) //cnt&1为真表示cnt位奇数
{
is_valid=false;
break;
}
cnt=0; //上一段连续的空格数量为偶数,再计算时归零
}
else cnt++; //该位为0,继续统计0的个数
}
if(cnt&1) is_valid=false;
st[i]=is_valid;
}
for(int j=0;j<1<<n;j++) //列举第j列所有状态
{
state[j].clear(); //测试的是多组样例,需要清空上一组样例存储的状态
for(int k=0;k<1<<n;k++) //列举第i-1列所有状态
{
if((j&k)==0&&st[j|k])
state[j].push_back(k);
}
}
//开始dp的过程
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]) //所有可以转移到j的合法方案
f[i][j]+=f[i-1][k];
cout<<f[m][0]<<endl;
}
return 0;
}