假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?
注意:给定 n 是一个正整数。
示例 1:
输入: 2
输出: 2
解释: 有两种方法可以爬到楼顶。
- 1 阶 + 1 阶
- 2 阶
使用迭代。一个for循环,因此时间复杂度为o(n)。
Go语言实现
func climbStairs(n int) int {
if n == 0{
return 0
}else if n == 1{
return 1
}else if n ==2{
return 2
}else{
a,b := 1,2
result := 0
for i := 3;i <= n;i++{
result = a+b
a,b = b,result
}
return result
}
}
python3实现
class Solution:
def climbStairs(self, n: int) -> int:
a,b=1,2
if n==1:
return 1
elif n==2:
return 2
elif n ==0:
return 0
else:
for i in range(3,n+1):
a,b = b,a+b
return b
python3简洁版
class Solution:
def climbStairs(self, n: int) -> int:
a,b=1,1
for i in range(1,n):
a,b=b,a+b
return b