徒步旅行中的补给问题 ｜ 豆包MarsCode AI 刷题

徒步旅行中的补给问题 ｜ 豆包MarsCode AI 刷题

在徒步旅行中，补给问题是一个常见的挑战。小R计划从地点A到地点B进行一次徒步旅行，总路程需要 ( N ) 天。为了确保每天都有足够的食物，小R每天必须消耗1份食物。幸运的是，小R每天都会经过一个补给站，可以购买食物进行补充。然而，每个补给站的食物价格可能不同，并且小R最多只能同时携带 ( K ) 份食物。小R希望在保证每天都有食物的前提下，以最小的花费完成这次徒步旅行。本文将探讨如何帮助小R计算最低的花费。

问题描述

小R的徒步旅行需要 ( N ) 天，每天必须消耗1份食物。每天经过的补给站的食物价格不同，且小R最多只能携带 ( K ) 份食物。目标是在保证每天都有食物的前提下，以最小的花费完成这次徒步旅行。

解决方案

为了帮助小R计算最低的花费，我们可以使用一个滑动窗口的方法来解决这个问题。具体步骤如下：

1. 初始化变量：
   • min_money：记录总花费，初始值为0。
   • ready：一个列表，用于存储当前窗口内的食物价格。
   • min_value：当前窗口内的最小食物价格。
2. 遍历每一天的食物价格：
   • 将当前天的食物价格添加到 ready 列表中。
   • 如果 ready 列表的长度超过 ( K )，则移除列表中的第一个元素，以保持窗口大小不超过 ( K )。
   • 计算 ready 列表中的最小值，并将其加到 min_money 中。
3. 返回结果：
   • 返回 min_money，即最低的花费。

代码实现

def solution(n, k, data):
    min_money = 0
    ready = []
    min_value = 0
    for i in data:
        ready.append(i)
        if len(ready) > k:
            ready.pop(0)
        min_value = min(ready)
        min_money += min_value
    return min_money
if __name__ == "__main__":
    # 测试样例
    print(solution(5, 2, [1, 2, 3, 3, 2]) == 9)  # 输出：True
    print(solution(6, 3, [4, 1, 5, 2, 1, 3]) == 9)  # 输出：True
    print(solution(4, 1, [3, 2, 4, 1]) == 10)  # 输出：True

代码解释

• 初始化：min_money 用于记录总花费，ready 用于存储当前窗口内的食物价格，min_value 用于存储当前窗口内的最小食物价格。
• 遍历每一天的食物价格：将当前天的食物价格添加到 ready 列表中。如果 ready 列表的长度超过 ( K )，则移除列表中的第一个元素，以保持窗口大小不超过 ( K )。计算 ready 列表中的最小值，并将其加到 min_money 中。
• 返回结果：返回 min_money，即最低的花费。

测试样例

• 样例1：n = 5, k = 2, data = [1, 2, 3, 3, 2]，输出：9
• 样例2：n = 6, k = 3, data = [4, 1, 5, 2, 1, 3]，输出：9
• 样例3：n = 4, k = 1, data = [3, 2, 4, 1]，输出：10 通过上述方法，我们可以有效地帮助小R计算出最低的花费，确保他在徒步旅行中每天都有足够的食物。