結果

問題 No.2114 01 Matching
コンテスト
ユーザー baluteshih
提出日時 2026-07-12 19:27:50
言語 C++23(gnu拡張gcc16)
(gcc 16.1.0 + boost 1.90.0)
コンパイル:
g++-16 -O2 -lm -std=gnu++23 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 331 ms / 5,000 ms
+ 644µs
コード長 5,713 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 4,516 ms
コンパイル使用メモリ 364,884 KB
実行使用メモリ 47,360 KB
最終ジャッジ日時 2026-07-12 19:28:11
合計ジャッジ時間 14,211 ms
ジャッジサーバーID
(参考情報)
judge2_0 / judge3_0
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 51
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

#line 1 "test2.cpp"
#define PROBLEM "https://yukicoder.me/problems/no/2114"
#line 2 "cplibrary/assumption.hpp"

#include <cassert>
#include <bits/stdc++.h>
#line 3 "test2.cpp"

#line 2 "cplibrary/DataStructure/Convex/ConjugateSlopeTrick.hpp"

#line 2 "cplibrary/DataStructure/Convex/SlopeTrick.hpp"

/*
reference: https://maspypy.com/slope-trick-1-解説編
*/

template<typename T, T INF = std::numeric_limits<T>::max() / 2>
class SlopeTrick {
    using min_heap = std::priority_queue<T, std::vector<T>, std::greater<T>>;
    using max_heap = std::priority_queue<T>;
    max_heap pq_l;
    min_heap pq_r;
    T min_f;
protected:
    T top_l() const { return pq_l.empty() ? -INF : pq_l.top() + add_l; }
    T top_r() const { return pq_r.empty() ? INF : pq_r.top() + add_r; }
    T pop_l() { T res = top_l(); if (!pq_l.empty()) pq_l.pop(); return res; }
    T pop_r() { T res = top_r(); if (!pq_r.empty()) pq_r.pop(); return res; }
    void push_l(T a) { pq_l.push(a - add_l); }
    void push_r(T a) { pq_r.push(a - add_r); }
    T add_l, add_r; 
public:
    SlopeTrick() : min_f(), add_l(), add_r() {}
    bool is_inf(T v) { return v >= INF || v <= -INF; }
    int size_l() { return pq_l.size(); }
    int size_r() { return pq_r.size(); }
    int size() { return size_l() + size_r(); }
    T top() { return min_f; }
    std::pair<T, T> top_interval() { return {top_l(), top_r()}; }
    SlopeTrick& add_const(T a) { return min_f += a, *this; }
    SlopeTrick& add_x_minus_a(T a) {
        min_f += std::max(T(0), top_l() - a);
        push_l(a), push_r(pop_l());
        return *this;
    }
    SlopeTrick& add_a_minus_x(T a) {
        min_f += std::max(T(0), a - top_r());
        push_r(a), push_l(pop_r());
        return *this;
    }
    SlopeTrick& add_abs(T a) { return add_x_minus_a(a).add_a_minus_x(a); } 
    SlopeTrick& add_linear(int a, T b) {
        for (min_f += b; a > 0; --a) {
            T x = pop_l();
            min_f += x, push_r(x);
        }
        for (; a < 0; ++a) {
            T x = pop_r();
            min_f -= x, push_l(x);
        }
        return *this;
    }
    SlopeTrick& shift(T a) { return add_l += a, add_r += a, *this; }
    SlopeTrick& sliding_window_minimum(T a, T b) {
        assert(a <= b);
        add_l += a, add_r += b;
        return *this;
    }
    SlopeTrick& prefix_min() { return min_heap().swap(pq_r), *this; }
    SlopeTrick& suffix_min() { return max_heap().swap(pq_l), *this; }
    T eval(T x) {
        T res = min_f;
        auto tl = pq_l; auto tr = pq_r;
        for (; !tl.empty(); tl.pop()) res += std::max(T(0), (tl.top() + add_l) - x);
        for (; !tr.empty(); tr.pop()) res += std::max(T(0), x - (tr.top() + add_r));
        return res;
    }
};
#line 4 "cplibrary/DataStructure/Convex/ConjugateSlopeTrick.hpp"

/*
reference: https://maspypy.com/slope-trick-3-slope-trick-の凸共役 
*/

template<typename T, T INF = std::numeric_limits<T>::max() / 2>
class ConjugateSlopeTrick : private SlopeTrick<T, INF> {
    using super = SlopeTrick<T, INF>;
public:
    ConjugateSlopeTrick() : super() {}
    using super::is_inf;
    T f0() { return -super::top(); }
    ConjugateSlopeTrick& add_const(T a) { return super::add_const(-a), *this; }
    ConjugateSlopeTrick& add_x_minus_a(T c, T a = 0) {
        if (a) shift(a);
        if (c > 0) super::add_r += c;
        if (c < 0) super::add_l += c;
        if (a) shift(-a);
        return *this;
    }
    ConjugateSlopeTrick& add_abs(T c, T a = 0) { return add_x_minus_a(c, a).add_x_minus_a(-c, a); }
    ConjugateSlopeTrick& add_linear(T a, T b = 0) { return super::shift(a).add_const(-b), *this; }
    ConjugateSlopeTrick& shift(int a) { return super::add_linear(-a, 0), *this; }
    ConjugateSlopeTrick& sliding_window_minimum(T a, T b) {
        assert(a <= b);
        shift(-a);
        for (int i = 0; i < b - a; ++i) super::add_x_minus_a(0);
        return *this;
    }
    ConjugateSlopeTrick& convolve(const std::vector<T>& g, int base = 0) {
        add_const(g[0]);
        for (int i = 1; i < int(g.size()); ++i) super::add_a_minus_x(g[i - 1] - g[i]);
        if (base != 0) shift(base);
        return *this;
    }
    ConjugateSlopeTrick& clear_left() { return super::suffix_min(), *this; }
    ConjugateSlopeTrick& clear_right() { return super::prefix_min(), *this; }
    T min_val(T p = 0) {
        return -super::eval(p);
    }
    T eval(T x = 0) {
        if (x > 0 && x > static_cast<T>(super::size_r())) return INF;
        if (x < 0 && -x > static_cast<T>(super::size_l())) return INF;
        if (x) shift(x);
        T res = f0();
        if (x) shift(-x);
        return res; 
    }
};
#line 5 "test2.cpp"

int main() {
    std::ios::sync_with_stdio(0), std::cin.tie(0);
    int n, m, k;
    std::cin >> n >> m >> k;
    long long ans = 0;
    std::map<int, std::vector<std::pair<int, int>>> arr;
    for (int i = 0; i < n; ++i) {
        int x;
        std::cin >> x;
        arr[x % k].emplace_back(x / k, n > m);
    }
    for (int i = 0; i < m; ++i) {
        int x;
        std::cin >> x;
        arr[x % k].emplace_back(x / k, n <= m);
    }
    for (auto [_, vec] : arr) {
        std::ranges::sort(vec);
        int tot = 0;
        ConjugateSlopeTrick<long long> slope;
        for (int i = 0; i < int(vec.size()); ++i) {
            if (vec[i].second == 1) {
                ++tot;
                slope.convolve({0, 0});
            }
            else {
                --tot;
                slope.shift(-1);
            }
            if (i + 1 < int(vec.size()))
                slope.add_abs(vec[i + 1].first - vec[i].first);
        }
        if (tot < 0) return std::cout << "-1\n", 0;
        ans += slope.f0();
    }
    std::cout << ans << "\n";
}
0