问题描述
小明最近迷上了积木画, 有这么两种类型的积木, 分别为 II 型(大小为 2 个单位面积) 和 LL 型 (大小为 3 个单位面积):
同时, 小明有一块面积大小为 2×N 的画布, 画布由 2×N 个 1×1 区域构成。小明需要用以上两种积木将画布拼满, 他想知道总共有多少种不同的方式? 积木可以任意旋转, 且画布的方向固定。
输入格式
输入一个整数 N,表示画布大小。
输出格式
输出一个整数表示答案。由于答案可能很大,所以输出其对 1000000007 取模后的值。
样例输入
3
样例输出
5
样例说明
五种情况如下图所示,颜色只是为了标识不同的积木:
评测用例规模与约定
对于所有测试用例,1≤N≤100000001≤N≤10000000.
代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef pair<int, int> PII;
const int N = 1e7, mod = 1e9 + 7;
LL dp[2][3]; // dp[i][j] 代表第i列已经放好,第i+1列已经放了j个格子
int n;
int main()
{
cin >> n;
dp[1][0] = 1; // 初始化第一列已放置0个格子的情况为1种
dp[1][1] = 2; // 初始化第一列已放置1个格子的情况为2种
dp[1][2] = 1; // 初始化第一列已放置2个格子的情况为1种
for (int i = 2; i <= n; i++)
{
// 计算第i列已经放置0个格子的情况数
dp[i & 1][0] = (dp[i - 1 & 1][0] + dp[i - 1 & 1][2] + dp[i & 1][1]) % mod;
// 计算第i列已经放置1个格子的情况数
dp[i & 1][1] = (dp[i - 1 & 1][0] * 2 + dp[i - 1 & 1][1]) % mod;
// 计算第i列已经放置2个格子的情况数
dp[i & 1][2] = dp[i - 1 & 1][0] % mod;
}
cout << dp[n & 1][0] << endl; // 输出第n列已经放置0个格子的情况数
return 0;
}