算法練習(36):Min Cost Climbing Stairs

題意:給出一個數組,可以從第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];
	}
};


發佈了80 篇原創文章 · 獲贊 72 · 訪問量 5萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章