LeetCode 746、使用最小花费爬楼梯
题目
题解
很显然这是一道动态规划题目,按照动态规划五步走:
-
确定dp数组以及下标的含义
dp[i] :到达第 i 个台阶所花费的最少体力
-
确定递推公式
dp[i] = min(dp[i - 1], dp[i - 2]) + cost[i];
-
dp数组如何初始化
dp[0] = cost[0]; dp[1] = cost[1];
-
确定遍历顺序
很显然是从前到后遍历cost数组
-
举例推导dp数组
… …
class Solution {
public:
int minCostClimbingStairs(vector<int>& cost) {
int len = cost.size();
vector<int> dp(len);
dp[0] = cost[0];
dp[1] = cost[1];
for (int i = 2; i < len; ++i) {
dp[i] = min(dp[i - 2], dp[i - 1]) + cost[i];
}
return min(dp[len - 1], dp[len - 2]);
}
};
- 时间复杂度:O(n)
- 空间复杂度:O(n)
优化一下空间复杂度,
class Solution {
public:
int minCostClimbingStairs(vector<int>& cost) {
int len = cost.size();
int dp[3];
dp[0] = cost[0];
dp[1] = cost[1];
for (int i = 2; i < len; ++i) {
dp[2] = min(dp[0], dp[1]) + cost[i];
dp[0] = dp[1];
dp[1] = dp[2];
}
return min(dp[0], dp[1]);
}
};