2020/05/06每日一题
找清楚子问题是什么,如何得到子问题下的解
dp思路的题目,典型特点在于寻找子问题的推导方法。对于这个题目子问题就是,days较少的时候,每次多增加一天都可以视为是新的dp。
- 状态:
dp[i]
第i天的花费,这里需要注意i天是实际的第i天。因此这一天可能需要出去旅游,也可能不需要旅游。 - 状态转移:考虑这一天是够需要出门旅游,如果需要,
dp[i] = min(dp[i-1]+costs[0], dp[i-7]+costs[1], dp[i-30]+costs[2])
;如果不需要,dp[i] = dp[i-1]
- 初始状态:因为没涉及到
dp[i]
和某值的比较,因此可以直接设为0.
class Solution:
def mincostTickets(self, days: List[int], costs: List[int]) -> int:
dic = set(days)
day = days[-1]
dp = [0]*(day+30)
for i in range(30, day+30):
if i-29 in dic:
dp[i] = min(dp[i-1]+costs[0], dp[i-7]+costs[1], dp[i-30]+costs[2])
else:
dp[i] = dp[i-1]
return dp[-1]
在代码中,为了简便直接多写了30位,这样可以保证dp[i-30]
不会出现越界。