結果

問題 No.2304 Distinct Elements
ユーザー rniyarniya
提出日時 2023-05-12 22:16:47
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 67 ms / 3,000 ms
コード長 3,989 bytes
コンパイル時間 2,586 ms
コンパイル使用メモリ 211,028 KB
実行使用メモリ 6,144 KB
最終ジャッジ日時 2023-08-19 05:08:12
合計ジャッジ時間 8,114 ms
ジャッジサーバーID
（参考情報） 		judge9 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,384 KB
testcase_01 AC 2 ms
4,380 KB
testcase_02 AC 1 ms
4,380 KB
testcase_03 AC 2 ms
4,380 KB
testcase_04 AC 43 ms
5,872 KB
testcase_05 AC 42 ms
5,828 KB
testcase_06 AC 42 ms
5,880 KB
testcase_07 AC 42 ms
6,004 KB
testcase_08 AC 42 ms
6,144 KB
testcase_09 AC 67 ms
5,948 KB
testcase_10 AC 66 ms
5,832 KB
testcase_11 AC 66 ms
5,864 KB
testcase_12 AC 66 ms
5,880 KB
testcase_13 AC 66 ms
5,872 KB
testcase_14 AC 2 ms
4,380 KB
testcase_15 AC 2 ms
4,380 KB
testcase_16 AC 2 ms
4,384 KB
testcase_17 AC 2 ms
4,380 KB
testcase_18 AC 1 ms
4,376 KB
testcase_19 AC 1 ms
4,384 KB
testcase_20 AC 1 ms
4,384 KB
testcase_21 AC 1 ms
4,376 KB
testcase_22 AC 2 ms
4,380 KB
testcase_23 AC 1 ms
4,380 KB
testcase_24 AC 2 ms
4,380 KB
testcase_25 AC 2 ms
4,376 KB
testcase_26 AC 2 ms
4,384 KB
testcase_27 AC 1 ms
4,380 KB
testcase_28 AC 1 ms
4,376 KB
testcase_29 AC 34 ms
5,640 KB
testcase_30 AC 4 ms
4,380 KB
testcase_31 AC 28 ms
4,660 KB
testcase_32 AC 52 ms
5,892 KB
testcase_33 AC 6 ms
4,380 KB
testcase_34 AC 54 ms
5,808 KB
testcase_35 AC 60 ms
5,764 KB
testcase_36 AC 4 ms
4,376 KB
testcase_37 AC 15 ms
4,376 KB
testcase_38 AC 45 ms
5,816 KB
testcase_39 AC 45 ms
5,872 KB
testcase_40 AC 26 ms
5,572 KB
testcase_41 AC 14 ms
4,392 KB
testcase_42 AC 35 ms
5,856 KB
testcase_43 AC 47 ms
6,000 KB
testcase_44 AC 47 ms
5,780 KB
testcase_45 AC 37 ms
4,604 KB
testcase_46 AC 29 ms
4,448 KB
testcase_47 AC 2 ms
4,380 KB
testcase_48 AC 2 ms
4,376 KB
testcase_49 AC 43 ms
5,836 KB
testcase_50 AC 44 ms
5,844 KB
testcase_51 AC 44 ms
5,832 KB
testcase_52 AC 39 ms
5,952 KB
testcase_53 AC 34 ms
5,944 KB
testcase_54 AC 37 ms
5,980 KB
testcase_55 AC 36 ms
5,964 KB
testcase_56 AC 35 ms
6,000 KB
testcase_57 AC 46 ms
6,056 KB
testcase_58 AC 44 ms
5,952 KB
testcase_59 AC 37 ms
5,940 KB
testcase_60 AC 2 ms
4,380 KB
testcase_61 AC 2 ms
4,388 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#ifdef LOCAL
#include <debug.hpp>
#else
#define debug(...) void(0)
#endif

template <typename T> struct SlopeTrick {
    // initialize as f(x) = 0
    SlopeTrick() : min_f(0), add_l(0), add_r(0) {}

    // argmin f(x), min f(x)
    std::tuple<T, T, T> query() const { return {top_L(), top_R(), min_f}; }

    // f(x) += b
    void add_const(const T& b) { min_f += b; }

    // f(x) += max(a - x, 0) \_
    void add_a_minus_x(const T& a) {
        min_f += std::max(T(0), a - top_R());
        push_R(a);
        push_L(pop_R());
    }

    // f(x) += max(x - a, 0) _/
    void add_x_minus_a(const T& a) {
        min_f += std::max(T(0), top_L() - a);
        push_L(a);
        push_R(pop_L());
    }

    // f(x) += |x - a| \/
    void add_abs(const T& a) {
        add_a_minus_x(a);
        add_x_minus_a(a);
    }

    // f(x) <- min_{y <= x} f(y) \/ -> \_
    void chmin_right() {
        while (!R.empty()) R.pop();
    }

    // f(x) <- min_{x <= y} f(y)
    void chmin_left() {
        while (!L.empty()) L.pop();
    }

    // f(x) <- min_{x - b <= y <= x - a} f(y)
    void shift(const T& a, const T& b) {
        assert(a <= b);
        add_l += a;
        add_r += b;
    }

    // f(x) <- f(x - a)
    void shift(const T& a) { shift(a, a); }

    // return f(x), f destructive
    T get_destructive(const T& x) {
        T res = min_f;
        while (!L.empty()) res += std::max(T(0), pop_L() - x);
        while (!R.empty()) res += std::max(T(0), x - pop_R());
        return res;
    }

    // f(x) += g(x), g destructive
    void merge_destructive(SlopeTrick& g) {
        if (g.size() < size()) {
            std::swap(min_f, g.min_f);
            std::swap(L, g.L);
            std::swap(R, g.R);
            std::swap(min_f, g.add_l);
            std::swap(min_f, g.add_r);
        }
        min_f += g.min_f;
        while (!g.L.empty()) add_a_minus_x(g.pop_L());
        while (!g.R.empty()) add_x_minus_a(g.pop_R());
    }

private:
    const T inf = std::numeric_limits<T>::max() / 2;
    T min_f;
    std::priority_queue<T, std::vector<T>, std::less<>> L;
    std::priority_queue<T, std::vector<T>, std::greater<>> R;
    T add_l, add_r;

    void push_L(const T& a) { L.emplace(a - add_l); }

    T top_L() const { return L.empty() ? -inf : L.top() + add_l; }

    T pop_L() {
        T res = top_L();
        if (!L.empty()) L.pop();
        return res;
    }

    void push_R(const T& a) { R.emplace(a - add_r); }

    T top_R() const { return R.empty() ? inf : R.top() + add_r; }

    T pop_R() {
        T res = top_R();
        if (!R.empty()) R.pop();
        return res;
    }

    size_t size() const { return L.size() + R.size(); }
};

using namespace std;

typedef long long ll;
#define all(x) begin(x), end(x)
constexpr int INF = (1 << 30) - 1;
constexpr long long IINF = (1LL << 60) - 1;
constexpr int dx[4] = {1, 0, -1, 0}, dy[4] = {0, 1, 0, -1};

template <class T> istream& operator>>(istream& is, vector<T>& v) {
    for (auto& x : v) is >> x;
    return is;
}

template <class T> ostream& operator<<(ostream& os, const vector<T>& v) {
    auto sep = "";
    for (const auto& x : v) os << exchange(sep, " ") << x;
    return os;
}

template <class T, class U = T> bool chmin(T& x, U&& y) { return y < x and (x = forward<U>(y), true); }

template <class T, class U = T> bool chmax(T& x, U&& y) { return x < y and (x = forward<U>(y), true); }

template <class T> void mkuni(vector<T>& v) {
    sort(begin(v), end(v));
    v.erase(unique(begin(v), end(v)), end(v));
}

template <class T> int lwb(const vector<T>& v, const T& x) { return lower_bound(begin(v), end(v), x) - begin(v); }

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int N;
    cin >> N;
    vector<int> A(N);
    cin >> A;

    sort(all(A));
    SlopeTrick<ll> ST;
    for (int i = 0; i < N; i++) {
        ST.chmin_right();
        ST.add_abs(A[i] - i);
    }

    ll ans = get<2>(ST.query());
    cout << ans << '\n';
    return 0;
}
0