5490. 吃掉 N 个橘子的最少天数
厨房里总共有 n 个橘子,你决定每一天选择如下方式之一吃这些橘子:
吃掉一个橘子。
如果剩余橘子数 n 能被 2 整除,那么你可以吃掉 n/2 个橘子。
如果剩余橘子数 n 能被 3 整除,那么你可以吃掉 2*(n/3) 个橘子。
每天你只能从以上 3 种方案中选择一种方案。
请你返回吃掉所有 n 个橘子的最少天数。
示例 1:
输入:n = 10
输出:4
解释:你总共有 10 个橘子。
第 1 天:吃 1 个橘子,剩余橘子数 10 - 1 = 9。
第 2 天:吃 6 个橘子,剩余橘子数 9 - 2*(9/3) = 9 - 6 = 3。(9 可以被 3 整除)
第 3 天:吃 2 个橘子,剩余橘子数 3 - 2*(3/3) = 3 - 2 = 1。
第 4 天:吃掉最后 1 个橘子,剩余橘子数 1 - 1 = 0。
你需要至少 4 天吃掉 10 个橘子。
- 橘子数量有四种情况
- 1.是2和3的倍数
- 2.是2的倍数,不是3的倍数
- 3.是3的倍数,不是2的倍数
- 4.既不是2的倍数,也不是3的倍数
这里使用的是离散化的dp,即自顶向下的动态规划加记录表搜索。我们在计算这几种情况的时候可以简化一下。
- 第一次判断如果是2的倍数,计算2的倍数。
- 第二次判断如果是3的倍数就计算3的倍数,注意由于之前已经求过一次2的倍数。假如该数是可以6的倍数,那么我们这里就不需要谈论既是2的倍数,也是3的倍数的情况了。
- 第三次判断是否是单个的数的倍数,或者都不是倍数。假如该数都不是2,3的倍数,那么需要和dfs(i-1)+1,取一个min. 注意,假如这里是2,或者3的倍数,也还是要和dfs(i-1)+1取一个min。为什么是6的倍数就可以不用计算与他前面小一的数的值,因为6的倍数,肯定是从2,3的倍数中间选一个值,肯定比i-1的值要小,所以没有必要。
from functools import lru_cache
class Solution:
def minDays(self, n: int) -> int:
memo = dict()
def dfs(i):
if i <= 2: return i
if i in memo:return memo[i]
ans = float('inf')
if i % 3 == 0:#是3的倍数
ans = min(ans, dfs(i//3)+1)
if i % 2 == 0:#是2的倍数
ans = min(ans, dfs(i//2)+1)
if i % 3 != 0 or i % 2 !=0:#不是2倍数,或者不是3的倍数,意思单独一个的倍数,或者都不是倍数
ans = min(ans, dfs(i-1)+1)
memo[i] = ans
return ans
return dfs(n)