LeetCode 746. 使用最小花費爬樓梯(C++、python)
阿新 • • 發佈:2018-12-14
陣列的每個索引做為一個階梯,第 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]
。
C
class Solution { public: int minCostClimbingStairs(vector<int>& cost) { int n=cost.size(); int temp[1001]; if(n==0) { return 0; } if(n==1) { return cost[0]; } if(n==2) { return min(cost[0],cost[1]); } temp[0]=0; temp[1]=0; for(int i=2;i<=n;i++) { temp[i]=min(temp[i-1]+cost[i-1],temp[i-2]+cost[i-2]); } return temp[n]; } };
python
class Solution: def minCostClimbingStairs(self, cost): """ :type cost: List[int] :rtype: int """ n=len(cost) if n==0: return 0 if n==1: return cost[0] if n==2: return min(cost[0],cost[1]) temp=[0 for i in range(1001)] temp[0]=0 temp[1]=0 for i in range(2,n+1): temp[i]=min(temp[i-1]+cost[i-1],temp[i-2]+cost[i-2]) return temp[n]