題意:給出一個數組,可以從第0位或第1位開始出發,停在哪個位就要花費那個數字的消耗,停了一個位置可以往後跳一步或者兩步,問走完整個數組所需最小花費。
分析與思路:這道題跟上一道題是一樣的思路的,動態方程有點相反而已,由於問題規定最多隻能走兩步,所以上一步沒走,當前就不能再不走,上一步走了,當前可以選擇走或者不走
所以動態規劃方程:dp[i][1]=max{dp[i-1][0]+cost[i],dp[i-1][1]+cost[i]},dp[i][0]=dp[i-1][1];最後取max{dp[i][1],dp[i][0]};
代碼:
class Solution {
public:
int minCostClimbingStairs(vector<int>& cost) {
if (cost.size() == 0) return 0;
vector<vector<int>> dp(cost.size(), vector<int>(2, 0));
dp[0][0] = 0;
dp[0][1] = cost[0];
for (int i = 1; i < cost.size(); i++) {
dp[i][0] = dp[i - 1][1];
dp[i][1] = dp[i - 1][0] + cost[i] > dp[i - 1][1] + cost[i] ? dp[i - 1][1] + cost[i] : dp[i - 1][0] + cost[i];
}
return dp[cost.size() - 1][0] > dp[cost.size() - 1][1] ? dp[cost.size() - 1][1] : dp[cost.size() - 1][0];
}
};