1. 程式人生 > >LeetCode 746. 使用最小花費爬樓梯(C++、python)

LeetCode 746. 使用最小花費爬樓梯(C++、python)

陣列的每個索引做為一個階梯,第 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]