結果

問題 No.2114 01 Matching
ユーザー 👑 rin204
提出日時 2022-10-29 14:15:39
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 334 ms / 5,000 ms
コード長 1,963 bytes
コンパイル時間 5,668 ms
コンパイル使用メモリ 298,048 KB
実行使用メモリ 47,232 KB
最終ジャッジ日時 2024-11-24 12:26:06
合計ジャッジ時間 17,223 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 51
権限があれば一括ダウンロードができます

ソースコード

diff #

#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#include<bits/stdc++.h>
#include<atcoder/lazysegtree>
using namespace std;
using namespace atcoder;

using S = long long;

S op(S l, S r){
    if(l <= r) return l;
    else return r;
}

S e(){
    return 1LL << 60;
}

using F = long long;

S mapping(F l, S r){
    return l + r;
}

F composition(F l, F r){
    return l + r;
}

F id(){
    return 0;
}

long long f(vector<int> &R, vector<int> &B){
    vector<pair<int, int>> event;
    for(auto r:R) event.push_back({r, 0});
    for(auto b:B) event.push_back({b, 1});
    sort(event.begin(), event.end());

    int n = R.size();
    int m = B.size();
    int z = n;
    lazy_segtree<S, op, e, F, mapping, composition, id> seg(n + m + 1);
    seg.set(z, 0);

    for(auto tmp:event){
        int x = tmp.first;
        if(tmp.second == 0){
            seg.apply(0, z, x);
            seg.apply(z, n + m + 1, -x);
            z--;
        }
        else{
            auto tmp = seg.get(z);
            z++;
            seg.apply(0, z, -x);
            seg.apply(z, n + m + 1, x);
            if(tmp < seg.get(z)){
                seg.set(z, tmp);
            }
        }
    }

    return seg.get(z);
}

void solve(){
    int n, m, k;
    cin >> n >> m >> k;
    map<int, vector<int>> B, R;
    int x;
    for(int i = 0; i < n; i++){
        cin >> x;
        B[x % k].push_back(x / k);
    }
    for(int i = 0; i < m; i++){
        cin >> x;
        R[x % k].push_back(x / k);
    }

    if(n > m){
        swap(n, m);
        swap(B, R);
    }

    long long ans = 0;
    
    for(auto tmp:B){
        int k = tmp.first;
        if(!R.count(k) || (B[k].size() > R[k].size())){
            cout << "-1\n";
            return;
        }
        ans += f(B[k], R[k]);
    }
    cout << ans << "\n";
    
}

int main(){
    cin.tie(0)->sync_with_stdio(0);
    int t;
    t = 1;
    // cin >> t;
    while(t--) solve();
}





0