刷题打卡：小U计划进行一场从地点A到地点B的徒步旅行，旅行总 - 是小刘 - 沸点 - 掘金

postgraduate @gzhu

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

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

M：总路程所需的天数。
N：路上补给站的数量。
p：每个补给站的描述，包含两个数字 A 和 B，表示第 A 天有一个补给站，并且该站每份食物的价格为 B 元。
保证第0天一定有一个补给站，并且补给站是按顺序出现的。
def solution(m:int, n:int, p:list)->int:
assert n == len(p) and m >= p[-1][0] and p[0][0] == 0
cost = 0
min_price, min_price_index = p[0][1], 0

for j in range(1, n):
a, b = p[j]
if b < min_price:
cost += min_price * (a - min_price_index)
min_price = b
min_price_index = a

cost += min_price * (m - min_price_index)
return cost

if __name__ == '__main__':
print(solution(m = 5 ,n = 4 ,p = [[0, 2], [1, 3], [2, 1], [3, 2]]) == 7)
print(solution(m = 6 ,n = 5 ,p = [[0, 1], [1, 5], [2, 2], [3, 4], [5, 1]]) == 6)
print(solution(m = 4 ,n = 3 ,p = [[0, 3], [2, 2], [3, 1]]) == 9)

展开

AI 刷题交流圈