leetcode 746.使用最小花费爬楼梯
题干
数组的每个索引作为一个阶梯,第 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]。
题解
很容易想到动态规划,因为到达每阶楼梯的最小体力花费都只和前两阶有关
class Solution {
public:
int minCostClimbingStairs(vector<int>& cost) {
int n = cost.size();
if(n == 1){
return cost[0];
}
vector<int> dp(n,0);
dp[0] = cost[0];
dp[1] = cost[1];
for(int i = 2 ; i < n ; ++i){
dp[i] = min(cost[i] + dp[i-1],cost[i] + dp[i-2]);
}
return min(dp[n-1],dp[n-2]);
}
};
既然只和前两阶有关,当然也可以用若干变量代替数组节省空间
class Solution {
public:
int minCostClimbingStairs(vector<int>& cost) {
int n = cost.size();
if(n == 1){
return cost[0];
}
int first = cost[0];
int second = cost[1];
for(int i = 2 ; i < n ; ++i){
int tempSecond = second;
second = min(second + cost[i],first + cost[i]);
first = tempSecond;
}
return min(first,second);
}
};