假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?
注意:给定 n 是一个正整数。
示例 1:
输入: 2
输出: 2
解释: 有两种方法可以爬到楼顶。
- 1 阶 + 1 阶
- 2 阶
示例 2:
输入: 3
输出: 3
解释: 有三种方法可以爬到楼顶。
- 1 阶 + 1 阶 + 1 阶
- 1 阶 + 2 阶
- 2 阶 + 1 阶
代码实现
C++:
从题目看的话,这是一个组合数学问题。于是我试着对n连续取不同的整数值,求不同的组合总数。
如下公式所示:
为了直观展示,我分别列出了当n取1,2,3,4,5,6时最终的不同走法组合总数。
结果看起来是不是很熟悉。这不是"斐波那契数列"吗?
我们简单分析,这个问题是否真的契合"斐波那契数列"。
斐波那契数列的递归形式可以用公式表示为:
Fib(x) = Fib(x-1)+Fib(x-2)
如果爬楼梯的问题契合"斐波那契数列"的话,就是说爬到第x台阶的所有方式,等于爬到第x-1台阶和第x-2台阶的所有方式的和。
答案是肯定的,理由是:
若我们先到达第x-1台阶我们只需再迈1个台阶就到达第x台阶;如果以这种方式,所有到达x层的方式就等于所有到达x-1层的方式。
若我们先到达第x-2台阶我们只需再一次迈2个台阶就到达第x台阶;如果以这种方式,所有到达第x台阶的方式就等于所有到达第x-2台阶的方式。
此外我们还要考虑到n的取值。
由于n值是正整数,且当n=1是,结果是1,所以可以转化为求Fib(n+1)的值的问题。
关于求斐波那契数列的问题,我们在"第68篇"说到过。这里我们依然是通过斐波那契数列的通项公式来求解。
class Solution {
public:
int climbStairs(int n) {
return (1/sqrt(5))*(pow((1+sqrt(5))/2, n+1) - pow((1-sqrt(5))/2, n+1));
}
};
Python:
class Solution(object):
def climbStairs(self, n):
"""
:type n: int
:rtype: int
"""
return int((1/(5**0.5))*(((1+5**0.5)/2)**(n+1) - ((1-5**0.5)/2)**(n+1)))