题目描述:有n阶楼梯,每次可以爬1或者2阶,问爬完n阶楼梯共有几种方案。
1.递归:
思路:假设现在在第x阶楼梯,他可以在x-1阶楼梯时再爬1阶到达x阶,也可以在x-2阶楼梯时再爬2阶到达x阶,设爬到x阶楼梯的总方案数为 f(x),则可以列出下面方程:(也可以一个个推理)
代码实现:(提交会超时)
class Solution {
public:
int climbStairs(int n) {
if(n<3){
return n; //1阶为1,2阶为2
}else{
return climbStairs(n-1)+climbStairs(n-2);
}
}
};
问题:因为这种方法会有好多重复递归部分,比如,要得到第5阶方案数,首先要得到4和3阶的方案数,然而
又要递归一次得到
,所以
就计算重复了。
优化思路:定义一个记忆数组,顾名思义,把递归过的结果储存在记忆数组中。
优化后代码:
class Solution {
public:
int climbStairs(int n) {
int a[9999999]={0}; //储存已递归的答案,初始化为0
return climbStairsT(n,a); //调用方法
}
int climbStairsT(int n,int a[]){
if(a[n]>0){ //不为0说明已经记忆过了,直接使用即可
return a[n];
}
else if(n<3){ //为1或为2阶时初始化
a[n]=n;
}else{ //没有记忆过,继续递归
a[n]=climbStairsT(n-1,a)+climbStairsT(n-2,a); //
}
return a[n];
}
};
2.动态规划:
思路:找到边界和状态转移方程,每阶楼梯的方案数表示为一个状态,用dp[x]表示,根据之前的思路,可得到状态转移方程为:
边界为第一和第二阶楼梯时
代码实现:
class Solution {
public:
int climbStairs(int n) {
int dp[999]; //定义dp数组
dp[1]=1; //边界
dp[2]=2;
for(int i=3;i<=n;i++){ //循环
dp[i]=dp[i-1]+dp[i-2]; //状态转移方程
}
return dp[n];
}
};
3.滚动数组思想:
思路:根据边界值,每次向后滚动,直到为n,直接上代码,应该好理解。
代码实现:
class Solution {
public:
int climbStairs(int n) {
int p=0,q=0,r=1; //初始值
for(int i=0;i<n;i++){ //循环
p=q; //向后滚动
q=r;
r=p+q;
}
return r;
}
};
这里最难理解的应该是p,q,r的初始值为什么是0,0,1,假如为0阶楼梯,方案数应该为1,所以r=1,然后自己模拟滚动试试看,就可以得到p,q的初始值了。
以上代码均可通过测试。