# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
332086 | LifeHappen__ | Roller Coaster Railroad (IOI16_railroad) | C++14 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
#include "railroad.h"
using namespace std;
#define ii pair<int, int>
#define fi first
#define se second
#define pb push_back
struct dsu {
vector <int> par;
int n;
dsu (int n) : n(n) {
par.assign(n, 0);
for (int i = 0; i < n; ++i) par[i] = i;
}
int findset(int u) {
if(par[u] == u) return u;
return par[u] = findset(par[u]);
}
bool mergeset(int u, int v) {
u = findset(u), v = findset(v);
if(u == v) return false;
par[v] = u; return true;
}
};
int64_t plan_roller_coaster(vector<int> s, vector<int> t) {
int n = s.size();
vector<int> vals;
for (int i = 0; i < n; ++i) {
vals.pb(s[i]); vals.pb(t[i]);
}
sort(begin(vals), end(vals));
vals.erase(unique(begin(vals), end(vals)));
dsu pa(vals.size() + 5);
vector<int> can(vals.size() + 5);
for (int i = 0; i < n; ++i) {
s[i] = lower_bound(begin(vals), end(vals), s[i]) - begin(vals);
t[i] = lower_bound(begin(vals), end(vals), t[i]) - begin(vals);
can[s[i]]++;
can[t[i]]--;
pa.mergeset(s[i], t[i]);
}
int m = vals.size();
int cur = 1;
int64_t ans = 0;
vector<tuple<int64_t, int, int>> ts;
for (int i = m - 1; i > 0; --i) {
cur += can[i];
if(cur < 0) {
ans += 1ll * (vals[i - 1] - vals[i]) * cur;
pa.mergeset(i - 1, i);
} else if(cur > 0) pa.mergeset(i - 1, i);
else ts.emplace_back(vals[i] - vals[i - 1], i, i - 1);
}
for (auto &_ : ts) {
int64_t w;
int u, v;
tie(w, u, v) = _;
if(pa.mergeset(u, v)) {
ans += w;
}
}
return ans;
}