- 爬楼梯
中文English
假设你正在爬楼梯,需要n步你才能到达顶部。但每次你只能爬一步或者两步,你能有多少种不同的方法爬到楼顶部?
样例
样例 1:
输入: n= 3
输出: 3
样例解释:
1) 1, 1, 1
2) 1, 2
3) 2, 1
共3种
样例 2:
输入: n = 1
输出: 1
解释:
只有一种方案
解题思路:
使用动态规划或者递归实现, 先列举所有特殊情况, 如: n =0 时, 只有 return 1, n =1 时, return 1, n =2 时, return 2
n =3时, 假设 第一次 走一步, 则 剩下 2步, 即 n =2,反之亦然。 因此 分析后, n = 3 时, 等于 n=1 + n =2 的和值。
计算公式:
f(0) = 0
f(1) = 1
f(2) = 2
f(3) = f(3-1) + f(3-2) = f(2) + f(1)
f(4) = f(4-1) + f(4-2) = f(3) + f(2)
因此: f(n) = f(n-1) + f(n-2)
代码如下:
class Solution:
"""
@param n: An integer
@return: An integer
"""
def climbStairs(self, n):
# write your code here
if n == 0:
return 1
if n == 1:
return 1
if n == 2:
return 2
else:
return self.climbStairs(n-1) + self.climbStairs(n-2)
以上代码在LeetCode 上提交通过, 但是在 LintCode 上提交显示超时,应该是 循环次数过多导致的, 时间复杂度不是最优,开始考虑优化。
继续上述的 解题思路,
计算公式:
f(0) = 0
f(1) = 1
f(2) = 2
f(3) = f(3-1) + f(3-2) = f(2) + f(1)
f(4) = f(4-1) + f(4-2) = f(3) + f(2)
f(5) = f(4) + f(3)
因此: f(n) = f(n-1) + f(n-2)
由于第一次代码的主要问题是 每次都要循环内多次, 如 n =4 时,扩展成 f(3) +f(2) , 但是f(3) 还需要继续循环, 导致耗时高,那么是否可以把之前的每次结果记录到一个列表中,每次取倒数第一和倒数第二的值 相加即可,这样不用持续循环,变成了 累加问题。
代码如下:
class Solution:
"""
@param n: An integer
@return: An integer
"""
def climbStairs(self, n):
# write your code here
if n == 0:
return 1
if n == 1:
return 1
if n == 2:
return 2
result = [1, 2] #起始结果
for i in range(n-2):
result.append(result[-1] + result[-2])
return result[-1]