Leetcode 983. 最低票價 (動態規劃)

// dp[i] 表示第i天的最少費用,如果這天不再計劃中,那麼dp[i] = dp[i-1],否則 dp[i] = min(dp[i-1]+cost[0], dp[i-7]+cost[1], dp[i-30]+cost[2]),這裏爲了處理越界,用max(0,i-30)處理即可。

class Solution {
public:
    int mincostTickets(vector<int>& days, vector<int>& costs) {
        // dp[i] 表示第i天旅行的最少費用。
        unordered_set<int> travelDays;
        for(auto day:days){
            travelDays.insert(day);
        }
        int lastDay = days.back();
        vector<int> dp(lastDay+1);
        for(int i=1;i<=lastDay;i++){
            if(!travelDays.count(i)){
                dp[i] = dp[i-1];
            }else{
                dp[i] = min(dp[max(0,i-1)]+costs[0],min(dp[max(0,i-7)]+costs[1],dp[max(0,i-30)]+costs[2])); 
            }
        }
        return dp[lastDay];
    }
};

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章