(39/60)DP基礎、斐波那契數、爬樓梯、用最小花費爬樓梯

Tazdingo發表於2024-03-08

  1. 爬到下標為i的位置所需最小花費為dp[i]
  2. dp[i] = min(dp[i-1] + cost[i-1],dp[i-2] + cost[i-2])
  3. dp[0] = 0; dp[1] = 0;
  4. left to right

複雜度分析

時間複雜度:O(N)。a

空間複雜度:O(N)。(狀態壓縮後為O(1))

程式碼實現

class Solution {
public:
    // 爬到下標為i的位置所需最小花費為dp[i]
    // dp[i] = min(dp[i-1] + cost[i-1],dp[i-2] + cost[i-2])
    // dp[0] = 0; dp[1] = 0;
    // left to right
    int minCostClimbingStairs(vector<int>& cost) {
        vector<int> dp(cost.size() + 1);
        dp[0] = 0; dp[1] = 0;
        for(int i = 2;i <= cost.size();i++) 
            dp[i] = min(dp[i-1] + cost[i-1],dp[i-2] + cost[i-2]);

        return dp.back();
    }
};

狀態壓縮版:

class Solution {
public:
    // 爬到下標為i的位置所需最小花費為dp[i]
    // dp[i] = min(dp[i-1] + cost[i-1],dp[i-2] + cost[i-2])
    // dp[0] = 0; dp[1] = 0;
    // left to right
    int minCostClimbingStairs(vector<int>& cost) {
        vector<int> dp(2);
        dp[0] = 0; dp[1] = 0;
        for(int i = 2;i <= cost.size();i++) {
            int sum = min(dp[1] + cost[i-1],dp[0] + cost[i-2]);
            dp[0] = dp[1];
            dp[1] = sum;
        }

        return dp[1];
    }
};

相關文章