Python3 三步爬樓梯問(wèn)題

原題地址 https://leetcode-cn.com/problems/three-steps-problem-lcci/

題目：

三步問(wèn)題。有個(gè)小孩正在上樓梯，樓梯有n階臺(tái)階，小孩一次可以上1階、2階或3階。實(shí)現(xiàn)一種方法，計(jì)算小孩有多少種上樓梯的方式。結(jié)果可能很大，你需要對(duì)結(jié)果模1000000007。

示例1:

輸入：n = 3 輸出：4說(shuō)明: 有四種走法

示例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# 本題用動(dòng)態(tài)規(guī)劃的思路解題，但是不需要n個(gè)數(shù)組空間，只需3個(gè)記錄e1 = 1e2 = 2e3 = 4for i in range(4, n+1):e4 = (e1 + e2 + e3)%1000000007 # 注意！取模一定要在這里就算好，而不是放到最后一步e1 = e2e2 = e3e3 = e4return e3

總結(jié)

以上是生活随笔為你收集整理的Python3 三步爬楼梯问题的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。