青蛙爬楼梯

题目描述

假设一只青蛙正在爬楼梯。需要 n 阶才能到达楼顶。

每次青蛙可以爬 1 或 2 个台阶。有多少种不同的方法可以爬到楼顶呢？

示例 1：
输入：n = 2
输出：2

解释：有两种方法可以爬到楼顶。
1. 1 阶 + 1 阶
2. 2 阶

示例 2：
输入：n = 3
输出：3

解释：有三种方法可以爬到楼顶。

1. 1 阶 + 1 阶 + 1 阶
2. 1 阶 + 2 阶
3. 2 阶 + 1 阶

算法思想

本题是非常典型的动态规划的题，台阶数量为1时，有一种方法，台阶数量为2时，有两种方法…
依次类推，此时可以动态分配一个大小为n+1的数组dp用来记录台阶数和方法数的对应关系。

由题干信息可知：当台阶数量大于等于3的时候，dp[i] = dp[i - 1] + dp[i - 2],即该项的值为后两项的和。此时利用一个for循环即可，最后返回dp[n]。

代码实现

class Solution {
public:
   int climbStairs(int n) {
        if (n <= 1)
            return 1;
        //开辟新内存空间
        int* dp = new int[n + 1];
        dp[1] = 1;
        dp[2] = 2;
        for (int i = 3; i <= n; i++) {
            dp[i] = dp[i - 1] + dp[i - 2];
        }
        return dp[n];
    }
};