斐波那契数列问题的递归和动态规划2

斐波那契数列问题的递归和动态规划2

题目描述

给出一个整数 n，代表台阶数，一次可以跨 2 个或者 1 个台阶，请输出有多少种走法。

输入描述：

第一行一个整数 n。

输出描述：

输出走法数对 1e9 + 7 取模的值。

示例1

输入

3

输出

3

备注：

$1\le n\le 10^{18}$

---

题解：

跟前面一题 斐波那契数列问题的递归和动态规划 基本一致，递推公式：F(n)=F(n-1)+F(n-2) ，唯一的不同就是 F(1)=1,F(2)=2。

代码：

#include <cstdio>
#include <cstring>

using namespace std;

typedef long long LL;
const LL MOD = 1e9 + 7;

LL tmp[2][2];
LL n;

void mat_mul(LL ret[][2], LL ans[][2]) {
    memset(tmp, 0LL, sizeof tmp);
    for (int i = 0; i < 2; ++i) {
        for (int j = 0; j < 2; ++j) {
            for (int k = 0; k < 2; ++k) {
                tmp[i][j] += ret[i][k] * ans[k][j] % MOD;
                tmp[i][j] %= MOD;
            }
        }
    }
    memcpy(ret, tmp, sizeof tmp);
}

void solve(LL n) {
    LL ret[][2] = {{1,0}, {0,1}};
    LL ans[][2] = {{1,1}, {1,0}};
    while (n) {
        if (n & 1LL) mat_mul(ret, ans);
        mat_mul(ans, ans);
        n >>= 1LL;
    }
    printf("%lld\n", (2LL * ret[0][0] % MOD + ret[1][0]) % MOD);
}

int main(void) {
    scanf("%lld", &n);
    if (n <= 3LL) return 0 * printf("%lld\n", n);
    solve(n - 2);
    return 0;
}