Python3 三步爬楼梯问题
生活随笔
收集整理的這篇文章主要介紹了
Python3 三步爬楼梯问题
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
Python3 三步爬樓梯問題
原題地址 https://leetcode-cn.com/problems/three-steps-problem-lcci/
題目:
三步問題。有個小孩正在上樓梯,樓梯有n階臺階,小孩一次可以上1階、2階或3階。實現一種方法,計算小孩有多少種上樓梯的方式。結果可能很大,你需要對結果模1000000007。
示例1:
輸入:n = 3 輸出:4說明: 有四種走法示例2:
輸入:n = 5輸出:13解題:
class Solution:def waysToStep(self, n: int) -> int:if n==0:return 0if n==1:return 1if n==2:return 2if n==3:return 4# 本題用動態規劃的思路解題,但是不需要n個數組空間,只需3個記錄e1 = 1e2 = 2e3 = 4for i in range(4, n+1):e4 = (e1 + e2 + e3)%1000000007 # 注意!取模一定要在這里就算好,而不是放到最后一步e1 = e2e2 = e3e3 = e4return e3總結
以上是生活随笔為你收集整理的Python3 三步爬楼梯问题的全部內容,希望文章能夠幫你解決所遇到的問題。