# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
660591 | Trisanu_Das | 전선 연결 (IOI17_wiring) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "wiring.h"
#include <bits/stdc++.h>
#define fi first
#define si second
using namespace std;
long long min_total_length(vector<int> r, vector<int> b) {
int n, m; n = r.size(); m=b.size();
vector<pair<int, int>> a;
for (auto &i:r) a.emplace_back(i, 1);
for (auto &i:b) a.emplace_back(i, -1);
sort(a.begin(), a.end());
reverse(r.begin(), r.end());
reverse(b.begin(), b.end());
int x = m;
long long r = INT_MIN, b = INT_MIN, ans = 0, s = 0;
vector<pair<long long, long long>> lst(n + m + 1, {INT_MAX, INT_MAX});
lst[m] = {0, 0};
for (auto &i:a) {
long long temp = 0;
(i.se == 1?r : b).pop_back();
x += i.se;
s += i.fi * i.se;
if (i.se == 1) {
r = i.fi;
temp = ans + i.fi - b;
if (b.size()) temp = min(temp, ans + b.back() - i.fi);
}
if (i.se == -1) {
b =i .fi;
temp = ans+i.fi-r;
if (r.size()) temp = min(temp, ans + r.back()-i.fi);
}
ans=min(temp, lst[x].fi + abs(lst[x].se - s));
lst[x] = {ans, s};
}
return ans;
}