增補部落格 第十九篇 python 爬樓梯

【題目描述】 假設一段樓梯共n(n>1)個臺階，小朋友一步最多能上3個臺階，那麼小朋友上這段樓梯一共有多少種方法。
【練習要求】請給出原始碼程式和執行測試結果，原始碼程式要求新增必要的註釋。
【輸入格式】在一行中輸入樓梯臺階的數目n。
【輸出格式】輸出小朋友上這段樓梯的方法數。
【輸入樣例】15
【輸出樣例】5768

def climbStairs(n):
    first3 = {1:1, 2:2, 3:4}
    if n in first3.keys():
        return first3[n]
    else:
        return climbStairs(n-1) + climbStairs(n-2) + climbStairs(n-3)

n = int(input("請輸入臺階數 n："))
ways = climbStairs(n)
print(f"小朋友上{n}個臺階的方法數為：{ways}")