积木画

问题描述

小明最近迷上了积木画, 有这么两种类型的积木, 分别为 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;
}