一,问题描述
二,问题分析
分析 n 层台阶有多少次方法 ,n层台阶的前一个状态是 n-1层,或者n-2层,所以 n 层台阶爬楼梯的方法数为 n-1 层的方法数+ n-2层的方法数,得出状态转移方程(递推公式): F(n) = F(n-1) + F(n-2),与斐波那契数列一致,所以递归和动态规划都是可以的
三,代码解答
1.递归算法
class Solution {
public:
int climbStairs(int n) {
if(n==1){
return 1;
}
else if(n==2){
return 2;
}
else return climbStairs(n-1) + climbStairs(n-2);
}
};
由于递归层次较多,计算了大量重复子问题,会超出时间限制,部分测试样例无法通过
2.记忆化搜索(递归算法的改进)
核心思想是空间换时间,用数组记录子问题的结果,避免子问题重复计算
class Solution {
private:
vector<int> memo; //备忘录,记录子问题的结果
int dfs(int n){
if(n==0 || n==1){
return 1;
}
if(memo[n] != -1){ //子问题已经被解决了,直接返回子问题的解
return memo[n];
}
int res = dfs(n-1) + dfs(n-2);
memo[n] = res;
return res;
}
public:
int climbStairs(int n) {
memo = vector<int> (n+1,-1); //数组初始化
return dfs(n);
}
};
3.动态规划
该问题的转移方程与斐波那契数列一样
class Solution {
public:
int climbStairs(int n) {
vector<int> memo(n+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];
}
};