题目:
70.爬楼梯
假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?
注意:给定 n 是一个正整数。
示例1:
输入: 2
输出: 2
解释: 有两种方法可以爬到楼顶。
1. 1 阶 + 1 阶
2. 2 阶
示例2:
输入: 3
输出: 3
解释: 有三种方法可以爬到楼顶。
1. 1 阶 + 1 阶 + 1 阶
2. 1 阶 + 2 阶
3. 2 阶 + 1 阶
解题思路:
爬到第n阶的方法为爬到第n-1阶和n-2阶的方法之和。因为爬到n-1阶后,再爬1阶就能到达n阶。爬到n-2阶后,可以一次性爬2阶到达n阶,也可以爬两个1阶到达n阶。因此只需要初始化爬到1阶和爬到2阶分别有多少种方法,便可以推导出爬到n阶的方法。
代码:
class Solution:
def climbStairs(self, n: int) -> int:
first = 1
second = 2
num = 0
for i in range(2, n):
num = first + second
first = second
second = num
return max(n, num)
运行结果: