题目解析 蛋糕工厂产能规划 | 豆包MarsCode AI刷题

38 阅读3分钟

问题描述

小明的蛋糕工厂每天可以生产的蛋糕数量是由工厂中的机器和工人的数量决定的,即 ( m \times w )。现在他收到了一个大订单,需要尽快生产出 ( n ) 个蛋糕。为了提升生产速度,小明可以使用每天生产的蛋糕去购买额外的机器或工人,每台机器或每个工人的成本是 ( p ) 个蛋糕。

例如,如果工厂起始时有 1 台机器和 2 个工人,每次扩大产能的成本是 1 个蛋糕,为了生产 60 个蛋糕,小明可以这样操作:

  • 第一天:生产 2 个蛋糕,买入 2 台机器(总机器数变为 3)。
  • 第二天:生产 6 个蛋糕,买入 3 台机器,3 个工人(机器数 6,工人数 5)。
  • 第三天:生产 30 个蛋糕。
  • 第四天:再生产 30 个蛋糕,完成订单。

你的任务是帮助小明计算最快多少天能完成订单。

思路分析

1. 贪心策略

我们首先考虑使用贪心策略来解决问题。每天结束时,我们将尽可能多地购买机器或工人,以最大化第二天的生产能力。为了平衡机器和工人的数量,我们优先增加数量较少的那种资源。

2. 动态规划

虽然贪心策略简单直观,但它可能不会总是找到最优解。因此,我们考虑使用动态规划的思想来优化解决方案。在每一步,我们不仅考虑当前的最优选择,还考虑未来可能的最优路径。

3. 循环与更新

我们将使用一个循环来模拟每天的生产和购买过程。在循环中,我们将更新总蛋糕数、机器数和工人数,并在每天结束时检查是否已经完成了订单。

算法设计

1. 初始化

我们初始化所需的天数 days,总蛋糕数 total_cakes,机器数 machines 和工人数 workers

2. 循环模拟

我们使用一个 for 循环来模拟每天的生产过程,直到总蛋糕数达到或超过订单要求。

3. 更新生产能力

每天结束时,我们计算可以购买的机器或工人的数量,并更新机器和工人数。我们尝试保持机器和工人数相等,以避免资源浪费。

4. 计算剩余天数

在购买新的机器或工人后,我们计算完成剩余蛋糕所需的天数,并更新所需的最少天数 days

代码实现

def solution(m, w, p, n):
    days = n // (m * w + 1) + 1
    total_cakes = 0
    machines = m
    workers = w
    for day in range(1, days + 1):
        total_cakes += machines * workers
        if total_cakes >= n:
            days = min(days, day)
            break
        added_machines_or_workers = total_cakes // p
        total_cakes -= added_machines_or_workers * p
        num = machines + workers + added_machines_or_workers
        machines = num // 2
        workers = num - machines
        days = min(days, day + ((n - total_cakes) // (machines * workers + 1) + 1))
    return days

复杂度分析

时间复杂度

算法的时间复杂度主要取决于循环的次数。在最坏的情况下,我们需要循环 O(n / (m * w)) 天,每天的操作包括常数时间的计算和更新。因此,总的时间复杂度为 O(n / (m * w))

空间复杂度

算法的空间复杂度为 O(1),因为我们只使用了常数个变量来存储中间结果。