补给站最优花费问题——贪心

48 阅读2分钟

题目地址:补给站最优花费问题 - MarsCode

问题描述

小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

解题思路

这道题的关键在于在第几天能够购买食物的最低价格是多少,也就是食物价格最低的那个补给站是哪个,比如第0天的补给站食物价格是这几天最低的,那么肯定一开始就将这几天的食物都买了花费最低,所以基本思路就是先初始化最低价格为第0天,然后遍历这几天,如果有补给站,看看是否食物价格更低,更低则更新最低价格,然后总花费再加上最低价。

代码

public static int solution(int m, int n, int[][] p) {
        int sum = 0;
        
        // 初始化当前最小价格为第一个补给站的价格
        int currentMinPrice = p[0][1];
        
        // 遍历每一天
        for (int day = 0; day < m; day++) {
            // 检查当前天数是否有补给站
            for (int i = 0; i < n; i++) {
                if (p[i][0] == day) {
                    // 更新当前最小价格
                    currentMinPrice = Math.min(currentMinPrice, p[i][1]);
                }
            }
            // 累加当天的食物花费
            sum += currentMinPrice;
        }
        
        return sum;