打卡第三天 | 豆包MarsCode AI刷题

123 阅读3分钟

补给站最优花费问题

问题描述

小U计划进行一场从地点A到地点B的徒步旅行,旅行总共需要 M 天。为了在旅途中确保安全,小U每天都需要消耗一份食物。在路程中,小U会经过一些补给站,这些补给站分布在不同的天数上,且每个补给站的食物价格各不相同。

小U需要在这些补给站中购买食物,以确保每天都有足够的食物。现在她想知道,如何规划在不同补给站的购买策略,以使她能够花费最少的钱顺利完成这次旅行。

  • M:总路程所需的天数。
  • N:路上补给站的数量。
  • p:每个补给站的描述,包含两个数字 A 和 B,表示第 A 天有一个补给站,并且该站每份食物的价格为 B 元。

保证第0天一定有一个补给站,并且补给站是按顺序出现的。


测试样例

样例1:

输入:m = 5 ,n = 4 ,p = [[0, 2], [1, 3], [2, 1], [3, 2]]
输出:7

样例2:

输入:m = 6 ,n = 5 ,p = [[0, 1], [1, 5], [2, 2], [3, 4], [5, 1]]
输出:6

样例3:

输入:m = 4 ,n = 3 ,p = [[0, 3], [2, 2], [3, 1]]
输出:9

解题思路

没有背包容量,直接购买可见范围内,最便宜的

代码

``python def solution(m: int, n: int, p: list[list[int]]) -> int:

    min_money = 0

    ready = []

    min_value = 0

    data = {[0]:[1] for _ in p}

    for i in range(m):

        # 当前站点加入ready

        # 所有可选的价格中最小的

        if i in data:

           ready.append(data[i])

        min_value = min(ready)

        min_money += min_value

    return min_money

if name == "main":

    # Add your test cases here

    print(solution(5, 4, [[0, 2], [1, 3], [2, 1], [3, 2]]) == 7) ``

徒步旅行中的补给问题

问题描述

小R正在计划一次从地点A到地点B的徒步旅行,总路程需要 N 天。为了在旅途中保持充足的能量,小R每天必须消耗1份食物。幸运的是,小R在路途中每天都会经过一个补给站,可以购买食物进行补充。然而,每个补给站的食物每份的价格可能不同,并且小R最多只能同时携带 K 份食物。

现在,小R希望在保证每天都有食物的前提下,以最小的花费完成这次徒步旅行。你能帮助小R计算出最低的花费是多少吗?


测试样例

样例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

解题思路

有背包容量,超过背包容量的要删除

代码

``python import heapq def solution(n, k, data): # Edit your code here min_money = 0 ready = [] min_value = 0 for i in data: # 当前站点加入ready ready.append(i) # 如果ready大于k,就将最先进入的站点价格删除 if len(ready) > k: ready.pop(0) # 找到最小值(这一步的时间复杂度应该为O(n))。如果在这里建堆就没必要了,时间复杂度可能提升至O(n*log()n) min_value = min(ready) print(min_value) min_money += min_value return min_money

if name == "main": # 添加你的测试用例 print(solution(5, 2, [1, 2, 3, 3, 2]) == 9) print(solution(6, 3, [4, 1, 5, 2, 1, 3]) == 9) print(solution(4, 1, [3, 2, 4, 1]) == 10) ``