自己提供一种更加简洁的写法,直接利用nums数组来存储我们整个DP过程中的最小值,每次遍历到i
时,nums[i]
中存放的就是满足题目限制条件,并且点亮第i
个烽火台的最小代价,然后我们把这个最优值加入队列。由于我们需要求出最后m大小的窗口里,最小的那个值,所以直接取此时单调队列对头的元素即可。
#include <iostream>
using namespace std;
const int N = 2e5 + 10;
int n, m;
int nums[N];
int q[N], head, tail;
int main()
{
cin >> n >> m;
for (int i = 1; i <= n; ++i)
cin >> nums[i];
head = 0, tail = 0;
for (int i = 1; i <= n; ++i)
{
nums[i] += nums[q[head]];
if (q[head] <= i - m) ++head;
while (head <= tail && nums[i] <= nums[q[tail]]) --tail;
q[++tail] = i;
}
cout << nums[q[head]] << endl;
return 0;
}