题目
在一个火车旅行很受欢迎的国度,你提前一年计划了一些火车旅行。在接下来的一年里,你要旅行的日子将以一个名为 days 的数组给出。每一项是一个从 1 到 365 的整数。
火车票有三种不同的销售方式:
一张为期一天的通行证售价为 costs[0] 美元;
一张为期七天的通行证售价为 costs[1] 美元;
一张为期三十天的通行证售价为 costs[2] 美元。
通行证允许数天无限制的旅行。 例如,如果我们在第 2 天获得一张为期 7 天的通行证,那么我们可以连着旅行 7 天:第 2 天、第 3 天、第 4 天、第 5 天、第 6 天、第 7 天和第 8 天。
返回你想要完成在给定的列表 days 中列出的每一天的旅行所需要的最低消费。
示例 1:
输入:days = [1,4,6,7,8,20], costs = [2,7,15]
输出:11
解释:
例如,这里有一种购买通行证的方法,可以让你完成你的旅行计划:
在第 1 天,你花了 costs[0] = $2 买了一张为期 1 天的通行证,它将在第 1 天生效。
在第 3 天,你花了 costs[1] = $7 买了一张为期 7 天的通行证,它将在第 3, 4, ..., 9 天生效。
在第 20 天,你花了 costs[0] = $2 买了一张为期 1 天的通行证,它将在第 20 天生效。
你总共花了 $11,并完成了你计划的每一天旅行。
示例 2:
输入:days = [1,2,3,4,5,6,7,8,9,10,30,31], costs = [2,7,15]
输出:17
解释:
例如,这里有一种购买通行证的方法,可以让你完成你的旅行计划:
在第 1 天,你花了 costs[2] = $15 买了一张为期 30 天的通行证,它将在第 1, 2, ..., 30 天生效。
在第 31 天,你花了 costs[0] = $2 买了一张为期 1 天的通行证,它将在第 31 天生效。
你总共花了 $17,并完成了你计划的每一天旅行。
提示:
1 <= days.length <= 365
1 <= days[i] <= 365
days 按顺序严格递增
costs.length == 3
1 <= costs[i] <= 1000
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/minimum-cost-for-tickets
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析(DP)
典型的DP题目,首先了解DP的基本思想:
将一个问题分解为n个小问题,分别为q1,q2,q3…qn,并且qn为该问题的最终解。
逐步求解q1,q2,q3…qn,每个小问题的输出都作为下一个小问题的输入,直到解出qn。
关键在于找到各个小问题中的输入和输出关系。
根据题意:
- 我们需要满足旅游日子数量的最低消费,第一反应应该是使用min函数进行计算。
- 我们需要计算的是
days
中,遍历到最后一个元素后的最低消费,从这个结论看来,符合DP的要求(qn为该问题的最终解) - 若想求最后一次旅行的最低消费qn,需要依次求得q(n - 1), q(n - 2)。我们需要找到每个小问题之间的关系。即再3种通行证中选择一种最低消费。
- 假设
dp
数组记录了旅游到每一天的最低消费,若dp[i]为第i天的最低消费,则dp[i]有三种可能: dp[i] = dp[i - 1] + const[0]
,这种情况是第i天买了1天的通行证。dp[i] = dp[i - 7] + const[1]
,这种情况是第i天买了7天的通行证,由于买的是7天的通行证,所以在第i - 7
到第i
天的消费都是一样的。dp[i] = dp[i - 30] + const[2]
,这种情况是第i天买了30天的通行证,与上条同理。- 最低消费为上述三种情况中最低消费,即
min(dp[i] = dp[i - 1] + const[0], dp[i - 7] + const[1], dp[i - 30] + const[2])
,需要注意的是i - 1
或i - 7
或i - 30
有可能会让数组越界,若是越界直接取第一天的消费即可。 dp
中没有旅游的天次中,最低消费与前一天相同。
在这个题中,我们可以看到第i
个解与第i - 1
,i - 7
,i - 30
有关系,将这层关系找到后就能解决该问题。
代码
class Solution {
public:
int mincostTickets(vector<int>& days, vector<int>& costs) {
vector<int> dp(days.back() + 1, 0);
// 记录在第几天旅游,用-1标识,没有旅游的天次用0标识。
for (auto &item : days) {
dp[item] = -1;
}
for (int i = 1; i < dp.size(); ++i) {
if (dp[i] == 0) {
dp[i] = dp[i - 1];
continue;
}
dp[i] = std::min({dp[i - 1] + costs[0], dp[max(0, i - 7)] + costs[1], dp[max(0, i - 30)] + costs[2]});
}
return dp.back(); // qn的解就是最终解
}
};