题目:
数组的每个索引作为一个阶梯,第 i个阶梯对应着一个非负数的体力花费值 costi。
每当你爬上一个阶梯你都要花费对应的体力花费值,然后你可以选择继续爬一个阶梯或者爬两个阶梯。
您需要找到达到楼层顶部的最低花费。在开始时,你可以选择从索引为 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]。
链接:https://leetcode-cn.com/problems/min-cost-climbing-stairs
思路:
1、本题可以用动态规划来进行解决。需要注意的是爬楼梯的话需要爬到顶楼,也就是说爬到数组外面,因此我们可以对数组添加一个元素0,计算爬到最后这个位置的最小话费
2、动态规划转移方程 dp[i] = min(dp[i-2]+cost[i-2], dp[i-1]+cost[i-1])
Python代码:
class Solution(object):
def minCostClimbingStairs(self, cost):
"""
:type cost: List[int]
:rtype: int
"""
dp = [0, 0]
cost.append(0)
size = len(cost)
if size<=3:
return 0
for i in range(2, size):
item = min(dp[i-2]+cost[i-2], dp[i-1]+cost[i-1])
dp.append(item)
return dp[size-1]
C++代码:
class Solution {
public:
int minCostClimbingStairs(vector<int>& cost) {
cost.push_back(0);
int size=cost.size();
if (size<=3) return 0;
vector<int> dp = {0, 0};
for (int i=2; i<size; i++){
int item = min(dp[i-2]+cost[i-2], dp[i-1]+cost[i-1]);
dp.push_back(item);
}
return dp[size-1];
}
};