983. Minimum Cost For Tickets

You have planned some train traveling one year in advance. The days of the year in which you will travel are given as an integer array days. Each day is an integer from 1 to 365.

Train tickets are sold in three different ways:

  • a 1-day pass is sold for costs[0] dollars,
  • a 7-day pass is sold for costs[1] dollars, and
  • a 30-day pass is sold for costs[2] dollars.

The passes allow that many days of consecutive travel.

  • For example, if we get a 7-day pass on day 2, then we can travel for 7 days: 2, 3, 4, 5, 6, 7, and 8.

Return the minimum number of dollars you need to travel every day in the given list of days.
 

Example 1:

Input: days = [1,4,6,7,8,20], costs = [2,7,15]
Output: 11
Explanation: For example, here is one way to buy passes that lets you travel your travel plan:
On day 1, you bought a 1-day pass for costs[0] = $2, which covered day 1.
On day 3, you bought a 7-day pass for costs[1] = $7, which covered days 3, 4, …, 9.
On day 20, you bought a 1-day pass for costs[0] = $2, which covered day 20.
In total, you spent $11 and covered all the days of your travel.

Example 2:

Input: days = [1,2,3,4,5,6,7,8,9,10,30,31], costs = [2,7,15]
Output: 17
Explanation: For example, here is one way to buy passes that lets you travel your travel plan:
On day 1, you bought a 30-day pass for costs[2] = $15 which covered days 1, 2, …, 30.
On day 31, you bought a 1-day pass for costs[0] = $2 which covered day 31.
In total, you spent $17 and covered all the days of your travel.

Constraints:
  • 1 <= days.length <= 365
  • 1 <= days[i] <= 365
  • days is in strictly increasing order.
  • costs.length == 3
  • 1 <= costs[i] <= 1000

From: LeetCode
Link: 983. Minimum Cost For Tickets


Solution:

Ideas:
  • Let dp[d] = minimum cost to cover all travel up to day d (1…365).

  • If day d is not a travel day: dp[d] = dp[d-1].

  • If day d is a travel day, buy one of 3 passes ending/covering d:

    • 1-day: dp[d] = dp[d-1] + costs[0]
    • 7-day: dp[d] = dp[max(0, d-7)] + costs[1]
    • 30-day: dp[d] = dp[max(0, d-30)] + costs[2]
  • Answer is dp[lastTravelDay] (or dp[365], both work).

Code:
static int min3(int a, int b, int c) {
    int m = (a < b) ? a : b;
    return (m < c) ? m : c;
}

int mincostTickets(int* days, int daysSize, int* costs, int costsSize) {
    (void)costsSize; // always 3 per problem

    int travel[366] = {0};   // travel[d] = 1 if we travel on day d
    int lastDay = days[daysSize - 1];
    for (int i = 0; i < daysSize; i++) {
        travel[days[i]] = 1;
    }

    int dp[366] = {0}; // dp[0] = 0

    for (int d = 1; d <= lastDay; d++) {
        if (!travel[d]) {
            dp[d] = dp[d - 1];
        } else {
            int d7  = (d - 7  >= 0) ? d - 7  : 0;
            int d30 = (d - 30 >= 0) ? d - 30 : 0;

            int cost1  = dp[d - 1] + costs[0];
            int cost7  = dp[d7]    + costs[1];
            int cost30 = dp[d30]   + costs[2];

            dp[d] = min3(cost1, cost7, cost30);
        }
    }

    return dp[lastDay];
}
Logo

AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。

更多推荐