青蛙跳台阶 Golang 与 Python 最简解法

「剑指Offer」里的经典题目，近期群里聊到这题，特来复习一波。

题目

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

思路

首先跳到 n 级台阶可以分解为两种情况：

1. 之前跳到 n-1 级台阶，然后再跳 1 级到达 n 级；
2. 之前跳到 n-2 级台阶，然后再跳 2 级到达 n 级；

F(n) = F(n-1) + F(n-2)

F(n-1) = F(n-2) + F(n-3)F(n-2) = F(n-3) + F(n-4)F(2) = F(1) + F(0)F(1) = 1F(0) = 1

F(0) + F(1) = F(2)F(n)

Golang 版

func JumpFloor(n int) int {
    a, b := 1, 1
    for ; n > 0; n-- {
        a, b = b, a + b
    }
    return a
}

Python 版

def jump_floor(n):
    a, b = 1, 1
    for _ in range(n):
        a, b = b, a + b
    return a