题解【AcWing1089】烽火传递

题面

单调队列优化 DP 模板题。

我们考虑设 (dp_{i}) 表示从 (1) 到 (i) 能够准确传递情报，且第 (i) 个烽火台发出信号的最小费用。

转移方程不难得出：(dp_{i} = min_{i-m leq j leq i-1}{dp_j} + a_i)（(a_i) 为第 (i) 个烽火台发出信号的费用）。

看到转移的条件：(i-m leq j leq i - 1)，这其实就是一个滑动窗口问题。

用一个单调队列维护长度为 (m) 的滑动窗口即可。

最后的答案是 (min_{n-m-1leq i leq n}{dp_i})。

#include <bits/stdc++.h>

using namespace std;

const int N = 200003;

int n, m, a[N], q[N], hh, tt, ans, dp[N];

int main()
{
    cin >> n >> m;
    for (int i = 1; i <= n; i+=1) cin >> a[i];
    hh = 0, tt = 0; //注意一开始队列中是有元素的
    for (int i = 1; i <= n; i+=1) //对于每一个烽火台进行转移
    {
        while (hh <= tt && q[hh] < i - m) ++hh; //队头是否在区间内
        dp[i] = dp[q[hh]] + a[i]; //DP 值转移
        while (hh <= tt && dp[q[tt]] >= dp[i]) --tt; //维护队列单调性
        q[++tt] = i; //加入队列
    }
    ans = 2000000007; //答案
    for (int i = n - m + 1; i <= n; i+=1) 
        ans = min(ans, dp[i]); //答案要从 n - m + 1 ~ n 的 DP 值中取最小值
    cout << ans << endl;
    return 0;
}