爬楼梯(go,python实现)

假设你正在爬楼梯。需要 n 阶你才能到达楼顶。

每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?

注意:给定 n 是一个正整数。

示例 1:

输入: 2
输出: 2
解释: 有两种方法可以爬到楼顶。

  1. 1 阶 + 1 阶
  2. 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

猜你喜欢

转载自blog.csdn.net/qq_45701131/article/details/109277143