题目链接:
题目描述:
小明最近迷上了积木画,有这么两种类型的积木,分别为 I 型(大小为 2 个单位面积)和 L 型(大小为 3 个单位面积):
同时,小明有一块面积大小为 2×N 的画布,画布由 2×N 个 1×1 区域构成。
小明需要用以上两种积木将画布拼满,他想知道总共有多少种不同的方式?
积木可以任意旋转,且画布的方向固定。
输入格式 输入一个整数 N,表示画布大小。
输出格式 输出一个整数表示答案。
由于答案可能很大,所以输出其对 1000000007 取模后的值。
数据范围 1≤N≤107。
输入样例:
3
输出样例:
5
样例解释:
五种情况如下图所示,颜色只是为了标识不同的积木:
题目分析:
本来还以为挺难的dp,跟蒙德里安的梦想一样,结果最后发现简单的很,因为题目规定了是2 X N,那么直接线性dp就可以了, 定义一个二维数组dp[N][3],第一维代表当前当前这一列已经放好了积木,第二维代表下一列突出来了几个格子,即:
- 当前这一列放好,下一列没有突出格子
dp[i][0] = dp[i - 1][0] + dp[i - 1][2] + dp[i - 2][1];(这一列可以放一根竖着的,也可以不放,也可以放反着的L) - 当前这一列放好,下一列突出一个格子
dp[i][1] = dp[i - 1][0] * 2 + dp[i - 1][1];(这一列可以放一根横着的,也可以放一根L)
3.当前这一列放好,下一列突出两个个格子
dp[i][2] = dp[i - 1][0]; (这一列必须放两根横着的,才能突出两个格子,那么上一列)
- 因为这题的数据范围比较大,如果开 dp[N][3] 的话可能空间就炸了,反正也只会用到前一个的状态,所以直接用滚动数组就可以了
算法1:
线性dp – 时间复杂度O(n):
代码:
/*
* @Author: suhuamo
* @Date: 2022-04-13 13:10:34
* @LastEditTime: 2022-04-13 15:23:03
* @FilePath: \algorithm\蓝桥杯\第十三届蓝桥杯省赛B\G.cpp
* @slogan: 也许散落在浩瀚宇宙的小行星们也知道
* 知识点: 线性dp
* 因为一共只有两行,所以可以直接开一个第二位包含3个状态的状态数组就可以了
* 状态可以划分为: 确保第i列已经摆放完成,第i+1列突出了几个格子
* 那么状态转移方程如下:
* dp[i & 1][0] = dp[i - 1 & 1][0] + dp[i - 1 & 1][2] + dp[i & 1][1];
dp[i & 1][1] = dp[i - 1 & 1][0] * 2 + dp[i - 1 & 1][1];
dp[i & 1][2] = dp[i - 1 & 1][0];
*/
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef pair<int, int> PII;
const int N = 1e7, Mod = 1e9 + 7;
int dp[2][3]; // dp[i][j] 代表第i列已经放好,第i+1列已经放了j个格子
int n;
int main()
{
cin >> n;
dp[1][0] = 1;
dp[1][1] = 2;
dp[1][2] = 1;
for(int i = 2; i <= n; i++)
{
dp[i & 1][0] = dp[i - 1 & 1][0] + dp[i - 1 & 1][2] + dp[i & 1][1];
dp[i & 1][1] = dp[i - 1 & 1][0] * 2 + dp[i - 1 & 1][1];
dp[i & 1][2] = dp[i - 1 & 1][0];
}
cout << dp[n & 1][0] << endl;
return 0;
}