이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
using namespace std;
using namespace __gnu_pbds;
using ll = long long;
using pi = pair<ll, pair<ll, ll>>;
using ii = pair<ll, ll>;
#define pb push_back
#define pp pop_back
#define ff first
#define ss second
#define lb lower_bound
typedef tree<int, null_type, less<int>, rb_tree_tag,
tree_order_statistics_node_update> indexed_set;
ll min_total_length(vector<int> r, vector<int> b) {
vector<ll> R, B;
for(auto u : r) R.pb(u);
for(auto u : b) B.pb(u);
ll N = R.size(), M = B.size(), res = 0;
vector<bool> vis(N, 0), seen(M, 0);
priority_queue<pi, vector<pi>, greater<pi>> pq;
for(ll l = 0; l < N; l++) {
for(ll i = 0; i < M; i++) {
pq.push({abs(R[l] - B[i]), {l, i}});
}
}
while(!pq.empty()) {
ll u = pq.top().ss.ff, v = pq.top().ss.ss, k = pq.top().ff; pq.pop();
if(vis[u] || seen[v]) continue;
res += k; vis[u] = seen[v] = 1;
}
for(ll l = 0; l < N; l++) {
if(vis[l]) continue;
auto it = lb(B.begin(), B.end(), R[l]);
ll ans = 1e18 + 7;
if(it != B.end()) ans = min(ans, abs(*it - R[l]));
if(it != B.begin()) {
it--; ans = min(ans, abs(*it - R[l]));
}
res += ans;
}
for(ll l = 0; l < M; l++) {
if(seen[l]) continue;
auto it = lb(R.begin(), R.end(), B[l]);
ll ans = 1e18 + 7;
if(it != R.end()) ans = min(ans, abs(*it - B[l]));
if(it != R.begin()) {
it--; ans = min(ans, abs(*it - B[l]));
}
res += ans;
}
return res;
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |