# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
906805 | mathnetic | Roller Coaster Railroad (IOI16_railroad) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <vector>
#include <algorithm>
using namespace std;
typedef long long ll;
ll plan_roller_coaster(const vector<int>& s, const vector<int>& t) {
unordered_map<int, int> uf;
auto fnd = [&](int x) {
if (uf[x] == 0) uf[x] = x;
return (x == uf[x]) ? x : (uf[x] = fnd(uf[x]));
};
auto un = [&](int x, int y) {
uf[fnd(x)] = fnd(y);
};
vector<pair<int, int>> e, ed;
int n = s.size();
for (int i = 0; i < n; ++i) {
un(s[i], t[i]);
e.emplace_back(s[i], 1);
e.emplace_back(t[i], -1);
}
sort(e.begin(), e.end());
int b = 0;
ll sol = 0;
for (int i = 0, j = 0; i < e.size(); i = j) {
for (; j < e.size() && e[j].first == e[i].first; ++j) {
b += e[j].second;
}
if (j < e.size()) {
if (b > 0) {
sol += static_cast<ll>(b) * (e[j].first - e[i].first);
}
if (b == 0) {
ed.emplace_back(e[i].first, e[j].first);
} else {
un(e[j].first, e[i].first);
}
}
}
sort(ed.begin(), ed.end(), [&](const auto& a, const auto& b) {
return (a.second - a.first) < (b.second - b.first);
});
for (auto& i : ed) {
if (fnd(i.first) != fnd(i.second)) {
sol += i.second - i.first;
un(i.first, i.second);
}
}
return sol;
}