假设你正在爬楼梯。需要 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 阶

动态规划

$f(n)$ 表示爬到第 $n$ 级台阶的方案数,根据题意,最后一步只有两种可能:跨了一级台阶,或者跨了两级台阶。

所以爬 $n$ 级台阶的方案总数是爬到第 $x−1$ 级台阶的方案数和爬到第 $x−2$ 级台阶的方案数的和。

可以列以下动态规划的转移方程:

$$f(n)=f(n−1)+f(n−2)$$

然后再考虑以下边界条件:

  1. 从第 0 级爬到第 0 级只有一种方案,$f(0) = 1$。
  2. 从第 0 级爬到第 1 级只有一种方案,$f(1) = 1$。

根据这两个边界条件再结合转移方程就可以推导出第 $n$ 级台阶的方案数。

递归解法就不写了,下面列出for循环解法和降低空间复杂度解法。


// for循环解法
class Solution {
public:
    int climbStairs(int n) {
        const int N = 1e5 + 10;
    	int dp[N] = {};
        dp[0] = 1;
        dp[1] = 1;
        for (int i = 2; i <= n; ++i) {
            dp[i] = dp[i - 1] + dp[i - 2];
        }
        return dp[n];
    }
};

// 由于 f(n) 只跟 f(n−1) 和 f(n−2) 有关,所以不需要长度为 n 的数组记录所有的方案数。
// 只需要 p 和 q 两个变量轮着记录就可以了。
class Solution {
public:
    int climbStairs(int n) {
        int p = 0, q = 0, r = 1;
        for (int i = 1; i <= n; ++i) {
            p = q; 
            q = r; 
            r = p + q;
        }
        return r;
    }
};

时间复杂度:循环执行 $n$ 次,每次花费常数的时间代价,故渐进时间复杂度为 $O(n)$。

空间复杂度:这里只用了常数个变量作为辅助空间,故渐进空间复杂度为 $O(1)$。

斐波那契数列通项公式

经过观察可以看出来方案数很像斐波那契数列,那么就可以直接用通项公式了。

主意斐波那契数列(0、1、1、2、3、5、8、13、21、34、……)和题目方案数列(1、1、2、3、5、8、13、21、34、……)开始的时候不太一样,所以在用主意斐波那契数列通项公式计算的时候, $n$ 需要加一。

推导过程如下:

根据线性递推数列方程 $f(n)=f(n−1)+f(n−2)$,可知道其特征方程为

$$x^2=x+1$$

解得 $$x_1 = {1 + \sqrt{5} \over 2},x_2 = {1 - \sqrt{5} \over 2}$$

则 $$f(n) = c_1x_1^n+c_2x_2^n$$

因为 $$f(1) = f(2) = 1$$

所以 $$c_1x_1+c_2x_2 = c_1x_1^2+c_2x_2^2 = 1$$

解得 $$c_1 = {1 \over \sqrt{5}},c_2 = -{1 \over \sqrt{5}},$$

所以 $$f(n) = {1 \over \sqrt{5}}[({1 + \sqrt{5} \over 2})^n - ({1 - \sqrt{5} \over 2})^n]$$

只要我们把这个公式转换成代码就可以求解了。

class Solution {
public:
    int climbStairs(int n) {
        double sqrt5 = sqrt(5);
        // 主意这里是 n + 1
        double fibn = pow((1 + sqrt5) / 2, n + 1) - pow((1 - sqrt5) / 2, n + 1);
        return (int)round(fibn / sqrt5);
    }
};

爬楼梯升级

条件改为每次你可以爬 $1$ 到 $n$ 个台阶。那么有多少种不同的方法可以爬到楼顶呢?

根据上题目的规律,可以得出下面公式:

$$f(n)=f(0)+f(1)+f(2)+…+f(n-2)+f(n-1)$$

因为 $$f(n-1)=f(0)+f(1)+f(2)+…f(n-2)$$

所以 $$f(n)=2f(n-1)$$

因为 $$f(0) = f(1) = 1$$

所以 $$f(2)=2f(1)=2,f(3)=2f(2)=4,f(4)=2f(3)=8$$

所以 $$f(n)=2^{n-1}$$

其实还有一个更巧妙的思路:一共要上 $n$ 个台阶,每次都可以爬 $1$ 到 $n$ 个台阶,算一共又多少种方法,那么到第 $n$ 个台阶的时候,把前面 $n-1$ 个台阶被踩和不被踩的可能性数量统计出来就是答案了,也就是 $2^{n-1}$。

代码过于简单,就不列出来了。