70. 爬楼梯

最新推荐文章于 2024-10-05 12:29:44 发布

吴天德少侠

最新推荐文章于 2024-10-05 12:29:44 发布

阅读量193

点赞数

文章标签： leetcode 动态规划

本文链接：https://blog.csdn.net/sdhdsf132452/article/details/121529991

版权

leetcode 专栏收录该内容

89 篇文章 0 订阅

订阅专栏

设 $f(n)$ 代表爬到第 $n$ 层的方法，因为你一次可以爬 $2$ 层或者 $1$ 层，所以你可以爬到 $f(n-2)$ 层再爬2层，或者爬到 $f(n-1)$ 层再爬1层。即

$f(n) = f(n-1) + f(n-2)\quad n>=3$
$f(1) = 1\quad f(2)=2$

cpp实现：

/*
 * @lc app=leetcode.cn id=70 lang=cpp
 *
 * [70] 爬楼梯
 */
#include<iostream>
#include<vector>
using namespace std;

// @lc code=start
class Solution {
public:
    int climbStairs(int n) {
        // f(n) = f(n-1)+f(n-2)
        
        if (n==1) return 1;  // 防止f[2]溢出
        // 开数组
        vector<int> f(n+1);
        f[1]=1, f[2]=2;
        for (int i=3; i<=n; i++){            
            f[i] = f[i-1]+f[i-2];
        }

        return f[n];
    }
};
// @lc code=end

由于 $f(n)$ 只于 $f(n-1)$ 和 $f(n-2)$ 有关，因此只需要3个整型变量即可将空间由 $O(n)$ 优化到 $O(1)$

class Solution {
public:
    int climbStairs(int n) {
        // f(n) = f(n-1)+f(n-2)
        // 开数组
        if (n==1) return 1;
       
        int pre=1, now=2;
        for (int i=3; i<=n; i++){            
            int next = pre+now;
            pre = now;
            now = next;
        }

        return now;
    }
};