数组的每个索引做为一个阶梯,第 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]
。
class Solution:
# dp[i] = min(dp[i- 2] + cost[i - 2], dp[i - 1] + cost[i - 1])
def minCostClimbingStairs(self, cost: 'List[int]') -> 'int':
length = len(cost)
dp = [0] * (length+1)
for i in range(2, length + 1):
dp[i] = min(dp[i - 2] + cost[i - 2], dp[i - 1] + cost[i - 1])
return dp[-1]
# dp[i] = cost[i] + min(dp[i- 1], dp[i - 2])
def minCostClimbingStairs2(self, cost: 'List[int]') -> 'int':
length = len(cost)
dp = [0] * (length + 1)
dp[0] = cost[0]
dp[1] = cost[1]
for i in range(2, length):
dp[i] = cost[i] + min(dp[i - 1], dp[i - 2])
return min(dp[length - 1], dp[length - 2])
#当前的dp值仅仅依赖前面两个的值,所以我们不必把整个dp数组都记录下来,只需用两个变量a和b来记录前两个值
def minCostClimbingStairs3(self, cost: 'List[int]') -> 'int':
a, b = 0, 0
for num in cost:
t = min(a, b) + num
a = b
b = t
return min(a, b)
if __name__=='__main__':
s = Solution()
test = ([10, 15, 20], [1, 2, 3], [1, 100, 1, 1, 1, 100, 1, 1, 100, 1], [1, 7, 3, 6, 5, 6], [1, 2, 3, 4, 5, 6])
for nums in test:
print(s.minCostClimbingStairs3(nums), end=' ')