斐波那契数列问题的递归和动态规划2
题目描述
给出一个整数 n,代表台阶数,一次可以跨 2 个或者 1 个台阶,请输出有多少种走法。
输入描述:
第一行一个整数 n。
输出描述:
输出走法数对 1e9 + 7 取模的值。
示例1
输入
3
输出
3
备注:
1 ≤ n ≤ 1 0 18 1 \leq n \leq 10^{18} 1≤n≤1018
题解:
跟前面一题 斐波那契数列问题的递归和动态规划 基本一致,递推公式: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;
}