力扣70:爬楼梯(动态规划和矩阵快速幂解法)

题目链接:https://leetcode.cn/problems/climbing-stairs/solutions/286022/pa-lou-ti-by-leetcode-solution/?envType=study-plan-v2&envId=top-100-liked

题目描述

假设你正在爬楼梯。需要 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 <= n <= 45

解法一(动态规划)

我们用dp[i]表示爬到第i级台阶的方案数,初始条件dp[1]=1,dp[2]=2考虑最后一步可能跨了一级台阶,也可能跨了两级台阶,所以我们可以列出如下式子:

dp[i]=dp[i-1]+dp[i-2]

上式即为动态转移方程它意味着爬到第i级台阶的方案数是爬到第i-1级台阶的方案数和爬到第i-2级台阶的方案数的和。很好理解,因为每次只能爬1级或2级,所以dp[i]只能从dp[i-1]dp[i-2]转移过来,而这里要统计方案总数,我们就需要对这两项的贡献求和。

我们不难通过转移方程和边界条件给出一个时间复杂度和空间复杂度都是O(n)的实现

代码
#include<iostream>
#include<vector>
#include <algorithm>
using namespace std;
class Solution {
public:
    int climbStairs(int n) {
        if (n == 1) return 1;
        if (n == 2) return 2;
        if (n >= 3) return climbStairs(n - 1) + climbStairs(n - 2);
        return 0;
    }
};
int main() {
    int n;
    cin >> n;
    Solution solution;
    int result = solution.climbStairs(n);
    cout << result;
    return 0;
}

由于这里的dp[i]只和dp[i-1]与dp[i-2]有关,所以我们可以用「滚动数组思想」把空间复杂度优化成 O(1)

优化后代码
class Solution {
public:
    int climbStairs(int n) {
        int p = 0, q = 0, r = 1;
        for (int i = 1; i <= n; ++i) {
            p = q; 
            q = r; 
            r = p + q;
        }
        return r;
    }
};

解法二(矩阵快速幂) 

通过数列关系我们可以构建这样一个递推过程:

\begin{gathered} {\left[\begin{array}{ll} 1 & 1 \\ 1 & 0 \end{array}\right]\left[\begin{array}{c} dp(n) \\ dp(n-1) \end{array}\right]=\left[\begin{array}{c} dp(n)+dp(n-1) \\ dp(n) \end{array}\right]=\left[\begin{array}{c} dp(n+1) \\ dp(n) \end{array}\right]} \\ {\left[\begin{array}{c} dp(n+1) \\ dp(n) \end{array}\right]=\left[\begin{array}{ll} 1 & 1 \\ 1 & 0 \end{array}\right]^n\left[\begin{array}{l} dp(1) \\ dp(0) \end{array}\right]} \\ \end{gathered}

M=\begin{bmatrix} 1 & 1\\ 1 & 0 \end{bmatrix}

因此我们只需要计算矩阵M的n次方即可得到dp(n)的值,时间复杂度为O(logn)

代码
class Solution {
public:
    vector<vector<long long>> multiply(vector<vector<long long>> &a, vector<vector<long long>> &b) {
        vector<vector<long long>> c(2, vector<long long>(2));
        for (int i = 0; i < 2; i++) {
            for (int j = 0; j < 2; j++) {
                c[i][j] = a[i][0] * b[0][j] + a[i][1] * b[1][j];
            }
        }
        return c;
    }

    vector<vector<long long>> matrixPow(vector<vector<long long>> a, int n) {
        vector<vector<long long>> ret = {{1, 0}, {0, 1}};
        while (n > 0) {
            if ((n & 1) == 1) {
                ret = multiply(ret, a);
            }
            n >>= 1;
            a = multiply(a, a);
        }
        return ret;
    }

    int climbStairs(int n) {
        vector<vector<long long>> ret = {{1, 1}, {1, 0}};
        vector<vector<long long>> res = matrixPow(ret, n);
        return res[0][0];
    }
};

参考链接:. - 力扣(LeetCode)

  • 31
    点赞
  • 26
    收藏
    觉得还不错? 一键收藏
  • 19
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 19
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值