LeetCode - 解题笔记 - 70 - Climbing Stairs

Solution 1

这道题和之前的求路径总数的思路其实很像，但是大大简化了。因为我们只有两种可能：一步走到和两步走到，因此我们只需要计算两个子问题。这两个子问题之前本身没有关联，因此状态转移方程为：

$$ans[i]=ans[i-1]+ans[i-2]$$

对于初始条件，我们需要提前设置好 $i$ 取1和2的情况（分别为1和2）。后续直接迭代就可以了。

• 时间复杂度： $O(n)$，其中 $n$为输出的数字，需要线性遍历一次所有的状态
• 时间复杂度： $O(n)$，其中 $n$为输出的数字，需要一个线性尺度的数组进行维护（其实完全可以简化到两个值进行维护）

仔细想想这个题，实际上就是斐波那契数列！

class Solution {
public:
    int climbStairs(int n) {
        if (n == 1) {
            return 1;
        }
        
        vector <int> ans = vector <int> (n + 1);
        ans[0] = 0;
        ans[1] = 1;
        ans[2] = 2;
        
        for (int i = 3; i <= n; ++i) {
            ans[i] = ans[i - 1] + ans[i - 2];
        }
        
        return ans[n];
    }
};

Solution 2

Solution 1的Python实现

class Solution:
    def climbStairs(self, n: int) -> int:
        if n == 1: return 1
        
        ans = [0] * (n + 1)
        ans[1] = 1
        ans[2] = 2
        
        for i in range(3, n + 1):
            ans[i] = ans[i - 1] + ans[i - 2]
        
        return ans[n]