补给站最优|豆包MarsCode AI刷题

101 阅读3分钟

这道题是关于小U进行徒步旅行时如何规划在不同补给站购买食物的策略问题。为了解决这个问题,我们可以定义一个二维动态规划(DP)数组来存储最小花费的信息。下面是对这个问题的详细解析:

问题理解

小U计划从地点A到地点B进行一次需要M天的徒步旅行。在旅途中,她每天都需要消耗一份食物。沿途有一些补给站,这些补给站分布在不同的天数上,每个补给站的食物价格不同。小U需要在补给站购买食物以确保每天都有足够的食物供应,同时希望找到一种策略使得总花费最少。

动态规划思路

我们使用一个二维数组dp来表示在第i天结束时拥有j份食物的最小花费。其中dp[i][j]代表前i天内恰好剩下j份食物时的最低费用。

初始化

首先,我们需要对dp数组进行初始化。由于保证第0天一定有一个补给站,且补给站是按顺序出现的,所以初始状态可以设置为所有天数都从第一个补给站购买所需数量的食物。

java

复制

for (int i = 0; i <= m; i++) {
    dp[0][i] = p[0][1] * i; // 第0天开始就有i份食物,每份价格为p[0][1]
}

状态转移方程

对于每一天i和每一份食物数量j,我们有以下几种情况:

  1. 不购买任何食物:这意味着我们从前一天的状态直接继承下来,即dp[i][j] = dp[i-1][j]

  2. 购买一些食物:如果当天存在补给站,那么可以考虑在该补给站购买一些食物以减少后续几天的开销。这里的关键是要决定购买多少份食物以及何时购买。

    java

    复制

    if (index < p.length && p[index][0] == i) {
        // 如果今天是补给站的一天
        for (int k = 0; k <= j; k++) {
            // 尝试购买k份食物
            dp[i][j] = Math.min(dp[i][j], dp[i-1][j-k+1] + k * p[index][1]);
        }
        index++; // 移动到下一个补给站
    } else {
        // 没有补给站,只能从前一天的状态继续
        dp[i][j] = dp[i-1][j];
    }
    

最终答案

最后,我们要找的是当第M天结束时至少剩下一份食物的最小花费,这可以通过访问dp[M][0]得到。

完整代码实现

以下是完整的Java代码实现:

java

复制

public class Main {
    public static int solution(int m, int n, int[][] p) {
        int[][] dp = new int[m+1][m+1];
        
        // 初始化第一天
        for (int i = 0; i <= m; i++) {
            dp[0][i] = p[0][1] * i;
        }

        int index = 0; // 补给站索引
        
        // 遍历每一天
        for (int i = 1; i <= m; i++) {
            // 遍历剩余食物的数量
            for (int j = 0; j < m; j++) {
                // 不购买食物的情况
                dp[i][j] = dp[i-1][j+1];

                // 如果当天有补给站
                if (index < p.length && p[index][0] == i) {
                    // 购买食物的情况
                    for (int k = 0; k <= j; k++) {
                        dp[i][j] = Math.min(dp[i][j], dp[i-1][j-k+1] + k * p[index][1]);
                    }
                    index++; // 移动到下一个补给站
                }
            }
        }
        
        return dp[m][0]; // 返回最终结果
    }
    
    public static void main(String[] args) {
        System.out.println(solution(5, 4, new int[][]{{0, 2}, {1, 3}, {2, 1}, {3, 2}})); // 应该返回7
    }
}

这段代码通过动态规划的方法解决了小U如何在补给站购买食物以使总花费最少的优化问题。