剑指offer 10. 变态跳台阶

原题

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

Reference Answer

解题思路:

依旧是典型回溯法，只需要找出递归规律即可，完事用回溯法实现。

递归规律为： res[n] = res[n-1] + res[n-2] + ...+ res[0] + 1，+1表示直接跳到第n级台阶。

# -*- coding:utf-8 -*-
class Solution:
    def jumpFloorII(self, number):
        # write code here
        res = []
        res.extend([0, 1, 2])
        # res[n] = res[n-1] + res[n-2] + ...+ res[0] + 1
        if number > 2:
            for i in range(3, number + 1):
                res.append(sum(res[:i]) + 1)
        return res[number]