青训营刷题一道题目

38 阅读1分钟

问题描述

小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

题目链接:徒步旅行中的补给问题 - MarsCode

分析如下:

image.png

代码实现

C++
 代码解读
复制代码
int solution(int n, int k, vector<int> data) {
    
    int res = 0, mini = 1e5 + 10; // res 记录最少钱,mini 记录当前最小值
    std::vector<int> ready; // 当前可携带食物价格
    //这里其实就像一个动态窗口!
    for (int i = 0; i < n; ++i) {
        // 当前站点加入ready
        ready.push_back(data[i]);
        // 如果ready大于k,就将最先进入的站点价格删除
        while (ready.size() > k) {
            ready.erase(ready.begin());
        }
        // 找到最小值
        mini = *min_element(ready.begin(), ready.end());
        res += mini;
        //cout << mini << " ";
    }
    return res;
}