假设你正在爬楼梯。需要 n 阶你才能到达楼顶。每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?
注意:给定 n 是一个正整数。
示例 1:
输入: 2
输出: 2
解释: 有两种方法可以爬到楼顶。
1. 1 阶 + 1 阶
2. 2 阶
示例 2:
输入: 3
输出: 3
解释: 有三种方法可以爬到楼顶。
1. 1 阶 + 1 阶 + 1 阶
2. 1 阶 + 2 阶
3. 2 阶 + 1 阶
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/climbing-stairs
方法:动态规划
这是一个重复子问题。爬n阶楼梯,那么我们要么爬一台阶要么两台阶。
所以首选动态规划,动态规划是自底向上的方法,所以我们需要知道初值。
n(1) = 1;
n(2) = n(1) + n(0) = 2 ;
故n(0) = 1;
这样初值设置完成。
我们采用一个记录函数,来避免重复计算。最后返回最后一个数字即可。
class Solution {
public:
int climbStairs(int n) {
vector<int> memo(n+1, -1);
memo[0] = 1;
memo[1] = 1;
for(int i = 2; i <= n; i++)
memo[i] = memo[i-1] + memo[i-2];
return memo[n];
}
};