289. 环路运输 - AcWing 滑动窗口维护环形dp

289. 环路运输 - AcWing

题意

一个环形公路上有 $n$ 个仓库，仓库之间的距离为两种走法的较小值，即 $dis(i,j)=min(|i-j|,n-|i-j|)$ 。从仓库 $i$ 到仓库 $j$ 的运输代价为 $a_i+a_j+dis(i,j)$ 。

求最大的运输代价。

解法

首先考虑把环展开成链，再复制一段。即 $a_{i+n}=a_i$。
$$\begin{gathered} a_i+a_j+dis(i,j) \\ =a_i+a_j+min(i-j,n-i+j),满足i\in [n+1,2n],j\in [i-n+1,i-1] \\ =\{\begin{array}{c}{a}_i+a_j+n-i+j,j\in [i-n+1,i-\frac{n}{2}-1]\\ a_i+a_j+i-j,j\in [i-\frac{n}{2},i-1]\end{array} \end{gathered}$$
所以我们可以从 $n+1$ 到$2n$ 遍历有端点 $i$ ，然后用滑动窗口来维护 $a_j+n-j$ 和 $a+j-j$ ，窗口的长度为 $\frac{n}{2}$ 。那么 $ans=ma{x}_{i=n+1}^{2n}(a_i-i+q{1}_{l1},a_i+i+q{2}_{l2})$ ，其中 $q{1}_{l1},q{2}_{l2}$ 分别表示两个滑动窗口内的最大值。

代码

#pragma region
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <vector>
using namespace std;
typedef long long ll;
#define tr t[root]
#define lson t[root << 1]
#define rson t[root << 1 | 1]
#define rep(i, a, n) for (int i = a; i <= n; ++i)
#define per(i, a, n) for (int i = n; i >= a; --i)
#pragma endregion
const int maxn = 2e6 + 5;
int n, a[maxn];
pair<int, int> q1[maxn], q2[maxn];  //单调递减
int l1 = 1, r1 = 0, l2 = 1, r2 = 0;
int main() {
    scanf("%d", &n);
    rep(i, 1, n) scanf("%d", &a[i]), a[i + n] = a[i];
    for (int i = 1; i < n / 2; ++i) {
        int p1 = i, p2 = i + n / 2;
        while (l1 <= r1 && q1[r1].second < a[p1] + p1 + n) --r1;
        q1[++r1] = {p1, a[p1] + p1 + n};
        while (l2 <= r2 && q2[r2].second < a[p2] - p2) --r2;
        q2[++r2] = {p2, a[p2] - p2};
    }
    int ans = 0;
    for (int i = n + 1; i <= 2 * n; ++i) {
        while (l1 <= r1 && q1[l1].first < i - n + 1) ++l1;
        while (l2 <= r2 && q2[l2].first < i - n / 2) ++l2;

        ans = max({ans, a[i] - i + q1[l1].second, a[i] + i + q2[l2].second});

        int p1 = i - 1 - n / 2, p2 = i - 1;
        while (l1 <= r1 && q1[r1].second < a[p1] + p1 + n) --r1;
        q1[++r1] = {p1, a[p1] + p1 + n};

        while (l2 <= r2 && q2[r2].second < a[p2] - p2) --r2;
        q2[++r2] = {p2, a[p2] - p2};
    }
    printf("%d\n", ans);
}