水果店果篮最小成本问题| 豆包MarsCode AI刷题

148 阅读4分钟

题目背景

小C开了一家水果店,某天接到了一个大订单,要求将 n 个编号为 1 到 n 的水果打包成多个果篮,每个果篮的最大容量为 m 个水果,并且果篮中水果的编号必须是连续的。每个果篮的成本与容纳的水果数量及水果的体积相关,成本公式为

k×⌊(u+v)/2⌋+s

其中: k 是果篮中水果的数量; u 是果篮中水果的最大体积; v 是果篮中水果的最小体积; s 是一个常数; ⌊x⌋ 表示对x 进行下取整。

我们的目标是帮助小C将这些水果打包成若干果篮,使得总成本最小。

题目分析

从题目描述中我们可以提炼出几个关键点:

  1. 连续性约束:每个果篮中的水果编号必须连续。这意味着我们在考虑果篮分配时,不能随意挑选水果,必须考虑它们的顺序。
  2. 最大容量约束:每个果篮最多只能装 m 个水果。这个限制使得我们无法将所有水果都打包到一个果篮中。
  3. 成本公式:果篮的成本不仅依赖于水果的数量,还依赖于水果的最大体积和最小体积。最大体积与最小体积的差异较大时,可能会导致较高的成本,因此我们需要合理地划分果篮,使得每个果篮中的水果体积差异尽量小。

解题思路

动态规划

这个问题可以通过 动态规划 来解决。我们的目标是最小化总成本,因此可以考虑构造一个动态规划数组 dp[i],表示将前 i 个水果分成若干果篮的最小成本。

状态转移

  • 初始化 dp[0] = 0,表示没有水果时,成本为零。
  • 对于每一个 i,我们需要考虑从前一个位置 j 开始,到当前位置 i 结束的一个果篮。这样我们可以枚举所有可能的划分方式,并通过计算每个划分的成本来更新 dp[i]
  • 具体地,状态转移的公式为: dp[i]=min(dp[j]+cost(j+1,i))dp[i]=min⁡(dp[j]+cost(j+1,i))
  • 其中,cost(j+1, i) 是将从第 j+1 到第 i 个水果作为一个果篮的成本。

计算单个果篮的成本

计算某个区间 [j+1, i] 的成本时,首先需要找到该区间的最大值 u 和最小值 v,然后使用给定的公式计算成本。这个过程可以通过遍历数组来实现。

优化

由于每个果篮最多可以包含 m 个水果,因此对于每个 i,我们只需要考虑从 i-mi 的划分,而不必考虑所有可能的前一个位置 j。这样可以减少计算量。

代码实现

image.png

代码详解

  1. dp[i] : 动态规划数组,dp[i] 表示从第一个水果到第 i 个水果的最小成本。
  2. max_volumemin_volume: 用于记录当前果篮中水果的最大和最小体积。
  3. cost(j+1, i) : 计算从 j+1i 这段水果的成本,公式为 k * ((max_volume + min_volume) // 2) + s,其中 k 为水果的数量,max_volumemin_volume 为该区间内的最大值和最小值。

复杂度分析

  • 时间复杂度:对于每一个 i,我们需要遍历从 i-mi 的区间,这样的操作最多进行 n 次,因此时间复杂度为 O(n×m)。
  • 空间复杂度:我们只需要一个长度为 n+1 的动态规划数组,因此空间复杂度为 O(n)。

优化方向

目前的实现采用了暴力枚举每个可能的果篮分割,时间复杂度为 O(n×m)O(n \times m)O(n×m)。如果水果数量 n 非常大,或者 m 值也较大,可以考虑以下几种优化方法:

  1. 提前计算最大最小值:对于每个区间 [j+1, i],可以通过滑动窗口等方法提前计算该区间的最大值和最小值,从而减少重复计算。
  2. 状态压缩:对于某些场景,可以尝试将 dp[i] 空间优化成较小的数组,减少空间开销。

总结

这个问题通过动态规划有效地解决了果篮划分和最小化总成本的挑战。在刷题过程中,我深刻体会到动态规划不仅需要理解状态和转移,还要对时间复杂度进行优化,避免暴力求解导致的时间超限。在以后的学习中,动态规划将会是一个常用且重要的工具,理解它的基本框架和优化技巧对解决许多类似问题至关重要。

对于刚开始接触动态规划的同学,我的建议是:首先理解每个状态的含义,再一步一步推导出状态转移公式,最后进行代码实现。在实现过程中,要特别注意状态转移的边界条件,并尽量避免不必要的重复计算。