【代码随想录day22】爬楼梯

小小的香辛料2023-08-17 13:23

题目

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

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

示例 1：

复制代码

输入：n = 2
输出：2
解释：有两种方法可以爬到楼顶。
1. 1 阶 + 1 阶
2. 2 阶

示例 2：

复制代码

输入：n = 3
输出：3
解释：有三种方法可以爬到楼顶。
1. 1 阶 + 1 阶 + 1 阶
2. 1 阶 + 2 阶
3. 2 阶 + 1 阶

思路

写法1

python 复制代码

class Solution:
    def climbStairs(self, n: int) -> int:
        if n==1:
            return 1
        # 1.确定dp数组及下标含义：dp[i]表示爬到i阶有dp[i]种方法
        # 2.状态转移方程
        # dp[i] = dp[i-1]+dp[i-2]:第i阶楼梯由第i-1阶和第i-2阶转移过来
        # 3.初始化：dp[1]=1,dp[2]=2
        # 4.执行顺序：顺序依次执行即可
        dp = [0]*(n+1)
        dp[1] = 1
        dp[2] = 2
        for i in range(3,n+1):
            dp[i] = dp[i-1]+dp[i-2]
        return dp[n]

写法2

python 复制代码

class Solution:
    def climbStairs(self, n: int) -> int:
        if n<3:
            return n
        dp1 = 1
        dp2 = 2
        res = 0
        for _ in range(n-2):
            res = dp1+dp2
            dp1,dp2 = dp2,res
        return res