掘金AI_徒步旅行补给问题

问题描述
小R正在计划一次从地点A到地点B的徒步旅行，总路程需要 N 天。为了在旅途中保持充足的能量，小R每天必须消耗1份食物。幸运的是，小R在路途中每天都会经过一个补给站，可以先购买完食物后再消耗今天的1份食物。然而，每个补给站的食物每份的价格可能不同，并且小R在购买完食物后最多只能同时携带 K 份食物。

现在，小R希望在保证每天食物消耗的前提下，以最小的花费完成这次徒步旅行。你能帮助小R计算出最低的花费是多少吗？

**输入 **

n 总路程需要的天数
k 小R最多能同时携带食物的份数
data[i] 第i天补给站每份食物的价格
**输出 **

返回完成这次徒步旅行的最小花费
**约束条件 **

1 < n,k < 1000
1 < data[i] < 10000
测试样例
样例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

#include <iostream>
#include <vector>
#include <deque>
/**
q.push(1);//将1压入队列
p.pop();//队首出队
q.front();//查询队首元素
q.back();//查询队尾元素
q.size();//查询队列中元素个数
q.empty();//若队列为空返回true

*/

using namespace std;

int solution(int n, int k, vector<int>& data) {
    // 使用双端队列来维护当前窗口内的最小值
    deque<pair<int, int>> mins;
    int result = 0;

    // n 是天数
    for (int j = 0; j < n; ++j) {
        // 维护单调队列：移除队尾比当前元素大的元素
        // 保证队列中的元素是递增的
        while (!mins.empty() && mins.back().second > data[j]) {
            mins.pop_back();  // 删除队尾
        }
        // 将当前元素加入队列
        mins.push_back({j, data[j]});
       
        // 移除超出窗口范围的元素（索引小于j-k+1）
        while (mins.front().first <= j - k) {
            mins.pop_front(); // 删除队头
        }
        
        // 当前窗口的最小值就是队首元素
        // 每天必须消耗1份食物，所以每天加上最小值
        result += mins.front().second;

        cout <<"每次购买的总价" << result << endl;
    }

    return result;
}

int main() {
    // 测试样例1
    vector<int> data1 = {1, 2, 3, 3, 2};
    cout << (solution(5, 2, data1) == 9) << endl;
    
    // 测试样例2
    vector<int> data2 = {4, 1, 5, 2, 1, 3};
    cout << (solution(6, 3, data2) == 9) << endl;
    
    // 测试样例3
    vector<int> data3 = {3, 2, 4, 1};
    cout << (solution(4, 1, data3) == 10) << endl;
    
    return 0;
}