爬樓梯（Python）

70. 爬樓梯

假設你正在爬樓梯。需要 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 階

def func(n):
    if n < 3:
        return n
    a, b, temp = 1, 2, 0
    for i in range(3, n + 1):
        temp = a + b
        a = b
        b = temp
    return temp

print(func(3))