Codeforces 1828D2 枚举

题意

传送门 Codeforces 1828D2 Range Sorting (Hard Version)

题解

先考虑简单情况，即 $n\le 5\cdot 10^3$，可以直接枚举 $O(n^2)$ 个子区间。固定左端点，拓展右端点的同时维护需要排序的段（可以通过求解新增元素在有序区间的插入位置做到）。总时间复杂度 $O(n^2\log n)$。

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
struct BIT {
    vector<int> a;
    BIT(int n) : a(n + 1) {}
    void add(int i, int d) {
        while (i < (int)a.size()) {
            a[i] += d;
            i += i & -i;
        }
    }
    int get(int i) {
        int s = 0;
        while (i > 0) {
            s += a[i];
            i -= i & -i;
        }
        return s;
    }
};

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int tt;
    cin >> tt;
    while (tt--) {
        int n;
        cin >> n;
        vector<int> a(n);
        for (int i = 0; i < n; ++i) {
            cin >> a[i];
        }
        auto xs = a;
        sort(xs.begin(), xs.end());
        xs.erase(unique(xs.begin(), xs.end()), xs.end());
        for (int &x : a) {
            x = lower_bound(xs.begin(), xs.end(), x) - xs.begin();
        }
        ll res = 0;
        for (int d = 0; d < n; ++d) {
            vector<pair<int, int>> ops;
            int cost = 0;
            BIT bit(xs.size());
            for (int i = 0; i + d < n; ++i) {
                int x = a[i + d];
                int pos = bit.get(x);
                if (pos != i) {
                    int r = i + 1;
                    int l = pos;
                    while (!ops.empty() && ops.back().second > l) {
                        l = min(l, ops.back().first);
                        cost -= ops.back().second - ops.back().first - 1;
                        ops.pop_back();
                    }
                    cost += r - l - 1;
                    ops.push_back({l, r});
                }
                res += cost;
                bit.add(x + 1, 1);
            }
        }
        cout << res << '\n';
    }

    return 0;
}

考虑 $n\le 3\cdot 10^5$ 的情况，此时难以直接枚举区间。由于排序开销是区间长度减一，需要尽可能地将区间分成尽可能多的段进行排序。对于 $[l,r)$ 可以分段的情况，存在一个位置 $k$，满足 ${\max }_{l\le i<k}{a}_i<{\min }_{k\le i<r}{a}_i$。枚举满足 ${\min }_{k\le i<r}{a}_i=a_j$ 的位置 $j$，找到其左右最邻近的满足 $a_x<a_j,a_y<a_j$ 的位置 $x,y$；观察到此时分界点 $k$ 仅可能为 $x+1$。令 $z$ 为最大的满足 $z<x,a_z>a_j$ 的位置。则可分段的区间 $[l,r)$ 满足 $z<l\le x,i<r\le y$。std::set 维护值域分别大于 / 小于 $a_j$ 的位置集合，总时间复杂度 $O(n\log n)$。

#include <bits/stdc++.h>
using namespace std;
using ll = long long;

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int tt;
    cin >> tt;
    while (tt--) {
        int n;
        cin >> n;
        vector<int> a(n);
        for (int i = 0; i < n; ++i) {
            cin >> a[i];
        }
        ll res = 0;
        for (int d = 1; d <= n; ++d) {
            res += (ll)(d - 1) * (n - d + 1);
        }
        vector<int> ord(n);
        iota(ord.begin(), ord.end(), 0);
        sort(ord.begin(), ord.end(), [&](int i, int j) {
            return a[i] < a[j];
        });
        set<int> big{-1, n}, small{-1, n};
        for (int i = 0; i < n; ++i) {
            big.insert(i);
        }
        for (int k = 0; k < n; ++k) {
            int i = ord[k];
            big.erase(i);

            auto it = small.upper_bound(i);
            int y = *it;
            int x = *prev(it);
            if (x != -1) {
                int z = *prev(big.lower_bound(x));
                res -= (ll)(y - i) * (x - z);
            }

            small.insert(i);
        }
        cout << res << '\n';
    }

    return 0;
}