【Leetcode】剑指Offer 10- II：青蛙跳台阶问题

一只青蛙一次可以跳上1级台阶，也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。 答案需要取模 1e9+7（1000000007），如计算初始结果为：1000000008，请返回 1。 示例 1： 输入：n = 2 输出：2 示例 2： 输入：n = 7 输出：21 示例 3： 输入：n = 0 输出：1 提示： 0 int: if n == 0: return 1 elif n == 1: return 1 else: def matrix(a, b): c = [[0, 0], [0, 0]] for i in range(2): for j in range(2): c[i][j] = int((a[i][0] * b[0][j] + a[i][1] * b[1][j]) % (1e9+7)) return c def power(a, n): ret = [[1, 0], [0, 1]] while n > 0: if n & 1: ret = matrix(ret, a) n >>= 1 a = matrix(a, a) return ret temp = power([[1, 1], [1, 0]], n-1) return int((temp[0][0] + temp[0][1]) % (1e9+7))

写的过程算是一个复习，不过快速幂写的依旧磕磕绊绊的，这个板子感觉还是可以简单地背一下。

妙解

class Solution:
    def numWays(self, n: int) -> int:
        a, b = 1, 1
        for _ in range(n):
            a, b = b, a + b
        return a % 1000000007

算是Python的一个特殊写法。果然写起来很优雅~