青训营-补给站最优花费问题

48 阅读2分钟

问题描述

小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天,最少要买支撑到下一个补给站数量的食物,若当前补给站食物价格比之前的价格高,我们选择从之前的补给站进行购买。

解题思路

我们用prize来标记之前的价格,给定prize=p[0][1]
首先,考虑特殊情况,只有一个补给站,从当前补给站购买m份食物即可。
其次,遍历p[i]数组,若当前补给站价格小于prize,更新prize,否则不更新;
购买可以支撑到下一个补给站数量的食物
最后,比较p[n-1][1]prize,进行更新,购买m-p[n-1][1]份食物

代码实现

public class Main {
    public static int solution(int m, int n, int[][] p) {
        // Edit your code here
        int result = 0;
        int prize = p[0][1];
        
        // 考虑特殊情况,即仅有一个补给站
        if (n == 1) {
            result = m * prize;
        } else {//更新prize
            result = prize * (p[1][0] - p[0][0]);
            for (int i = 1; i < n - 1; i++) {
                if (p[i][1] < prize) {
                    prize = p[i][1];
                }
                result += prize * (p[i + 1][0] - p[i][0]);
            }
            if(p[n-1][1]<prize) prize = p[n-1][1];
            result += prize * (m - p[n - 1][0]);
        }
       
        return result;
        
    }

    public static void main(String[] args) {
        // Add your test cases here

        System.out.println(solution(5, 4, new int[][]{{0, 2}, {1, 3}, {2, 1}, {3, 2}}) == 7);
    }
}