数组的每个索引做为一个阶梯,第 i个阶梯对应着一个非负数的体力花费值 cost[i](索引从0开始)。
每当你爬上一个阶梯你都要花费对应的体力花费值,然后你可以选择继续爬一个阶梯或者爬两个阶梯。
您需要找到达到楼层顶部的最低花费。在开始时,你可以选择从索引为 0 或 1 的元素作为初始阶梯。
示例 1:
输入: cost = [10, 15, 20]
输出: 15
解释: 最低花费是从cost[1]开始,然后走两步即可到阶梯顶,一共花费15。
示例 2:
输入: cost = [1, 100, 1, 1, 1, 100, 1, 1, 100, 1]
输出: 6
解释: 最低花费方式是从cost[0]开始,逐个经过那些1,跳过cost[3],一共花费6。
注意:
cost 的长度将会在 [2, 1000]。
每一个 cost[i] 将会是一个Integer类型,范围为 [0, 999]。
解法:
注意动态规划方程:dp[n] = min(dp[n-1] + num,dp[n-2]+num)
dp[n-1] 是cur_min,dp[n-2]是pre_min,num是cos[n]
注意
10 15 20 和 0 0 1 0
因为到倒数第二台阶可以直接跳,或者到最后一个台阶才跳
所以要比较最后dp[n]和dp[n-1]哪个更小
# -*- coding: utf-8 -*-
class Solution(object):
def minCostClimbingStaire(self,cost):
cur_min = 0
pre_min = 0
for i in range(len(cost)):
tmp = cur_min
cur_min = min(cur_min + cost[i],pre_min + cost[i])
pre_min = tmp
if i == len(cost) -2:
min1 = cur_min
if i == len(cost) -1:
min2 = cur_min
return min(min1,min2)
if __name__ == "__main__":
s = Solution()
cost = list(map(int,input().split()))
print(s.minCostClimbingStaire(cost))