結果

問題 No.2114 01 Matching
ユーザー yamake
提出日時 2022-10-28 23:16:34
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,361 bytes
コンパイル時間 2,416 ms
コンパイル使用メモリ 207,104 KB
最終ジャッジ日時 2025-02-08 15:04:41
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 27 WA * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;
#define rep(i,n) for(int i=0;i<n;++i)
#define rep1(i,n) for(int i=1;i<=n;++i)
#define rrep(i,n) for(int i=n-1;i>=0;--i)
#define debug(output) if(debugFlag)cout<<#output<<"= "<<output<<"\n";
using lint = long long;
typedef pair<int,int> P;
const bool debugFlag=true;
const lint linf=1.1e18;const int inf=1.01e9;
constexpr int MOD=1000000007;
template<class T>bool chmax(T &a, const T &b) { if(a <= b){ a = b; return 1; } return 0; }
template<class T>bool chmin(T &a, const T &b) { if(a > b){ a = b; return 1; } return 0; }

void solve(vector<int>& a,vector<int>& b,int k){
    int n=a.size();int m=b.size();
    map<int,multiset<int>> mp;
    rep(i,m)mp[b[i]%k].insert(b[i]);
    lint res=0;
    sort(a.begin(),a.end());
    rep(i,n){
        auto& v=mp[a[i]%k];
        auto it=v.lower_bound(a[i]);
        int mn=inf;
        if(it!=v.end())chmin(mn,(*it-a[i])/k);
        if(it!=v.begin()){
            --it;
            if(chmin(mn,(a[i]-*(it))/k))v.erase(it);
            else{
                ++it;
                if(it!=v.end())v.erase(it);
            }
        }
        res+=mn;
    }
    if(res>=inf)res=-1;
    cout<<res<<"\n";
}

signed main(){
    int n,m,k;cin>>n>>m>>k;
    vector<int> a(n),b(m);
    rep(i,n)cin>>a[i];
    rep(i,m)cin>>b[i];
    if(n>m)swap(a,b);
    solve(a,b,k);
    return 0;
}
0