动态规划, 每次走的步骤取决于前两个的状态; 状态转移方程: 当长度 < 2 时,直接返回 cost[0]; 当长度 >= 2 时,当前值取决于min(dp[i-1], dp[i-2]) + cost[i]; 代码如下:
class Solution:
def minCostClimbingStairs(self , cost: List[int]) -> int:
# write code here
if len(cost) == 1: return cost[0]
n = len(cost)
dp = [cost[0], cost[1]]
for i in range(2, n):
dp.append(min(dp[i - 1], dp[i - 2]) + cost[i])
return min(dp[-1], dp[-2])