【算法】变态跳台阶

一、题目

一只青蛙一次可以跳上1级台阶，也可以跳上2级……它也可以跳上n级。求该青蛙跳上一个n级的台阶总共有多少种跳法。

二、解题思路

n级台阶的跳法有n种情况：

• 最后一跳跳了一级
• 最后一跳跳了两级
• ……
• 最后一跳跳了n级

所以，n级台阶的跳法就相当于n-1级再跳一次一级台阶+n-2级再跳一次二级台阶+…+0级最后跳一次n级台阶的跳法之和，由此可得出以下递推公式：

$$f(n)=\{\begin{array}{ll}0,& n=0\\ 1,& n=1\\ f(n-1)+f(n-2)+\dots +f(0),& n>1\end{array}$$

当$n>1$时：
$$\begin{array}{rl}f(n)& =f(n-1)+f(n-2)+\dots +f(0)\\ f(n-1)& =f(n-2)+\dots +f(0)\end{array}$$所以，
$$\begin{array}{rl}f(n)& =f(n-1)+(f(n-2)+\dots +f(0))\\ & =f(n-1)+f(n-1)\\ & =2f(n-1)\end{array}$$因此，递推公式可进一步简化为：
$$f(n)=\{\begin{array}{ll}1,& n=1\\ 2f(n-1),& n>1\end{array}$$

分析到这里不难发现本题的本质与斐波那契数列问题相似。因此解题的方法也基本相同。

三、解题方法

代码实现使用 Python 2.7.3。

（一）递归

我们用递归算法就能够轻易实现以上式子，时间复杂度为$O(n)$。

# -*- coding:utf-8 -*-
class Solution:
    def jumpFloorII(self, number):
        if number == 1:
            return 1
        else:
            return 2 * self.jumpFloorII(number-1)

（二）遍历

我们考虑对于任意一个$N$，我们从第一项开始计算，并将所有的中间结果保存下来，这样在计算$f(n)$时，只需要通过查表直接获取$f(n-1)$的值计算即可，时间复杂度为$O(n)$。

# -*- coding:utf-8 -*-
class Solution:
    def jumpFloorII(self, number):
        if number == 1:
            return 1
        else:
            array = [1]
            for i in range(1, number):
                array.append(2*array[i-1])
            return array[-1]

（三）快速乘法

实际上，遍历的方法还不是最优的，本题最优的解决方法的时间复杂度只需要$O(\log n)$。

当$n>1$时：
$$\begin{array}{rl}f(n)& =2f(n-1)\\ & =2^{2}f(n-2)\\ & =\dots \dots \\ & =2^{n-1}f(1)\end{array}$$这样，本问题就变成了如何用最快的方法求$2^n$的问题。

快速求$2^{11}$的过程如下：

1. 11的二进制形式为1011
2. $2^{11}=2^{1\times 8}\times 2^{0\times 4}\times 2^{1\times 2}\times 2^{1\times 1}$

在这个过程中，我们先求出$2^1$，然后根据$2^1$求出$2^2$，再根据$2^2$求出$2^4$，…，最后求出$2^8$。因此，11的二进制数形式总共多少位，就使用了几次乘法

然后把应该累乘的值相乘即可，即$2^8$、$2^2$、$2^1$应该累乘，因为8、2、1对应到11的二进制数中，相应位上是1；而$2^4$不应该累乘，因为4对应到11的二进制数中，相应位上是0。

# -*- coding:utf-8 -*-
class Solution:
    def jumpFloorII(self, number):
        if number == 1:
            return 1
        else:
            multiplier = 2
            res = 1
            
            times = number - 1
            while times:
                if times&1:
                    res *= multiplier
                    
                multiplier *= multiplier
                
                times = times >> 1
                
            return res