LeetCode：746. Min Cost Climbing Stairs - Python

问题描述：

746. 使用最小花费爬楼梯
数组的每个索引做为一个阶梯，第 i个阶梯对应着一个非负数的体力花费值 cost[i](索引从 0开始)。

每当你爬上一个阶梯你都要花费对应的体力花费值，然后你可以选择继续爬一个阶梯或者爬两个阶梯。

您需要找到达到楼层顶部的最低花费。在开始时，你可以选择从索引为0 或 1的元素作为初始阶梯。

示例 :

输入: 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[i] = min(dp[i-1]+cost[i-1], dp[i-2]+cost[i-2])  # 要考虑楼顶哦

Python3实现：

# @Time   :2018/08/08
# @Author :LiuYinxing
# 动态规划


class Solution:
    def minCostClimbingStairs(self, cost):
        n = len(cost)
        if n < 2: return 0
        if n == 0: return min(cost)
        dp = [0] * (n+1)
        for i in range(2, n+1):
            dp[i] = min(dp[i-1]+cost[i-1], dp[i-2]+cost[i-2])
        return dp[n]

    def minCostClimbingStairs1(self, cost):  # 压缩空间
        n = len(cost)
        if n < 2: return 0
        if n == 0: return min(cost)
        dp1, dp2 = 0, 0
        for i in range(2, n+1):
            cur = min(dp1+cost[i-1], dp2+cost[i-2])
            dp2 = dp1
            dp1 = cur
        return cur


if __name__ == '__main__':
    cost = [1, 100, 1, 1, 1, 100, 1, 1, 100, 1]
    solu = Solution()
    print(solu.minCostClimbingStairs(cost))

欢迎指正哦。