蓝桥杯2022积木画_DP

P2037 - [蓝桥杯2022初赛] 积木画 - New Online Judgehttp://oj.ecustacm.cn/problem.php?id=2037

当n=3时的5种情况为：

用dp[i]记录铺满2*i的画布的方案数，最终答案输出dp[n]即可

用moreOne[i]记录铺满2*i的画布后，第i+1列多出一个积木的方案数

dp[i] = dp[i - 1] + dp[i - 2] + moreOne[i - 2]

分别对应在dp[i-1]的基础上竖着放一个I型积木

和在dp[i-2]的基础上横着放两个I型积木

和在moreOne[i-2]的基础上放一个L型积木

moreOne[i] = dp[i - 1] * 2 + moreOne[i - 1]

分别对应在dp[i-1]的基础上补一个L型积木（可以上下翻转所以*2）

和在moreOne[i-1]的基础上横着放一个I型积木（moreOne[i-1]已包含翻转的情况所以不*2）

#include <iostream>
using namespace std;
typedef long long ll;
const int N = 1e7 + 4, C = 1e9 + 7;
ll dp[N] = {0}, moreOne[N] = {0};
int main() {
	int n;
	cin >> n;
	dp[0] = 1;
	moreOne[0] = 0;
	dp[1] = 1;
	moreOne[1] = 2;
	for (int i = 2; i <= n; i++) {
		dp[i] = (dp[i - 1] + dp[i - 2] + moreOne[i - 2]) % C;
		moreOne[i] = (dp[i - 1] * 2 + moreOne[i - 1]) % C;
	}
	cout << dp[n];
	return 0;
}

---

另解：如果不记录moreOne的状态，也可以改成记录从dp[0]到dp[i-3]的和

 

dp[i] = dp[i - 1] + dp[i - 2] + 2 * sum[i - 3]

其中dp[i-1] + dp[i-2]和上一种做法的意义相同，除了这两种之外，

如图1、2，还需要在dp[i-3]的基础上加上由两个L型积木拼成的情况，（上下翻转）

如图3，还需要在dp[i-4]的基础上加上两个L型积木+一个I型积木的情况，（上下翻转要*2）

如图4，还需要在dp[i-5]的基础上加上右下两个L型积木+两个I型积木的情况，（上下翻转要*2）

......

相当于在两个L型积木中间加上k个I型积木所得出的长度为3+k的方案数，

所以记录从dp[0]到dp[i-3]的和即可

#include <iostream>
using namespace std;
typedef long long ll;
const int N = 1e7 + 4, C = 1e9 + 7;
ll dp[N] = {0}, sum[N] = {0};
int main() {
    int n;
    cin >> n;
    dp[0] = sum[0] = 1;
    dp[1] = 1;
    dp[2] = 2;
    for (int i = 3; i <= n; i++) {
        dp[i] = (dp[i - 1] + dp[i - 2] + 2 * sum[i - 3]) % C;
        sum[i - 2] = (sum[i - 3] + dp[i - 2]) % C;
    }
    cout << dp[n];
    return 0;
}

可以对空间复杂度进行优化：

#include <iostream>
using namespace std;
typedef long long ll;
const int N = 1e7 + 4, C = 1e9 + 7;
int main() {
    int n;
    cin >> n;
    ll cur, pre = 2, prev = 1, sum = 1;
    for (int i = 3; i <= n; i++) {
        cur = (pre + prev + 2 * sum) % C;
        sum = (sum + prev) % C;
        prev = pre;
        pre = cur;
    }
    cout << cur;
    return 0;
}