描述
数组的每个索引作为一个阶梯,第 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]。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/min-cost-climbing-stairs/
求解
class Solution {
private:
vector<int> record; // 存储数据,避免递归中重复计算, record[i]表示到第i阶楼梯的最低消耗
int help(const vector<int> &cost, int i) {
if (i == 0 || i == 1) {
return record[i];
}
if (record[i] == INT_MAX) {
record[i] = std::min(help(cost, i - 1), help(cost, i - 2)) + cost[i];
}
return record[i];
}
public:
// 递归求解
int minCostClimbingStairs_recur(const vector<int> &cost) {
if (cost.empty()) {
return 0;
}
// 初始化
record.resize(cost.size(), INT_MAX);
record[0] = cost[0];
if (cost.size() > 1) {
record[1] = cost[1];
}
// 问题可转换为到n-1阶和n-2阶楼梯的最小消耗,之后均可一步或者两步跨入顶层
return std::min(help(cost, cost.size() - 1), help(cost, cost.size() - 2));
}
// 非递归,动态规划,空间复杂度O(N)
int minCostClimbingStairs(const vector<int> &cost) {
if (cost.empty()) {
return 0;
}
// 初始化
vector<int> dp;
dp.resize(cost.size(), INT_MAX);
dp[0] = cost[0];
if (cost.size() > 1) {
dp[1] = cost[1];
}
const int N = cost.size();
for (int i = 2; i < N; ++i) {
dp[i] = std::min(dp[i - 1], dp[i - 2]) + cost[i];
}
return std::min(dp[N - 1], dp[N - 2]);
}
// 非递归,动态规划,空间复杂度O(1)
int minCostClimbingStairs(const vector<int> &cost) {
if (cost.empty()) {
return 0;
}
const int N = cost.size();
// 初始化
int pre = 0; // 在计算到达台阶i的时候,已经迭代计算出的到达台阶i-1的耗费
int curr = 0; // 在计算到达台阶i的时候,已经迭代计算出的到达台阶i-2的耗费
int res = pre;
for (int i = 2; i <= N; ++i) {
// 计算到达台阶i的最小耗费(注意,到达台阶i并不包含台阶i的耗费)
res = std::min(pre + cost[i-2], curr + cost[i-1]);
pre = curr;
curr = res;
}
return res;
}
};