LeetCode 70.爬楼梯(递归、动态规划,滚动数组思想实现)

题目描述:有n阶楼梯,每次可以爬1或者2阶,问爬完n阶楼梯共有几种方案。

1.递归:

思路:假设现在在第x阶楼梯,他可以在x-1阶楼梯时再爬1阶到达x阶,也可以在x-2阶楼梯时再爬2阶到达x阶,设爬到x阶楼梯的总方案数为 f(x),则可以列出下面方程:(也可以一个个推理)

f(x)=f(x-1)+f(x-2)

代码实现:(提交会超时)

class Solution {
public:
    int climbStairs(int n) {
        if(n<3){
            return n; //1阶为1,2阶为2
        }else{
            return climbStairs(n-1)+climbStairs(n-2);
        }
    }
};

问题:因为这种方法会有好多重复递归部分,比如f(5)=f(4)+f(3),要得到第5阶方案数,首先要得到4和3阶的方案数,然而f(4)=f(3)+f(2)又要递归一次得到f(3),所以f(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]表示,根据之前的思路,可得到状态转移方程为:

dp(x)=dp(x-1)+dp(x-2)

边界为第一和第二阶楼梯时 f(1)=1  f(2)=2

代码实现:

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的初始值了。

以上代码均可通过测试。

  • 2
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值