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;
}