#include <bits/stdc++.h>
#include "railroad.h"
#define ll long long
#define vt vector
#define pb push_back
#define ar array
#define all(x) (x).begin(), (x).end()
#define sz(x) (x).size()
using namespace std;
/*
1. simplify
2. add new elements
3. brute force solution
4. optimize
5. start implementing
*/
// --- templates ---
// --- code ---
vt<int> p;
int find(int u){
if(p[u] == u){
return u;
}
return p[u] = find(p[u]);
}
bool join(int u, int v){
if((u = find(u)) == (v = find(v))){
return false;
}
p[u] = v;
return true;
}
long long plan_roller_coaster(std::vector<int> s, std::vector<int> t) {
int n = sz(s);
vt<int> comp = s;
comp.insert(comp.end(), all(t));
sort(all(comp));
vt<int> a(sz(comp));
p.resize(sz(comp));
iota(all(p), 0);
for(int i = 0; i < n; i++){
s[i] = lower_bound(all(comp), s[i]) - comp.begin();
t[i] = lower_bound(all(comp), t[i]) - comp.begin();
a[s[i]]++;
a[t[i]]--;
join(s[i], t[i]);
}
ll ans = 0;
vt<ar<int, 2>> e;
for(int i = 0; i < sz(comp) - 1; i++){
a[i + 1] += a[i];
ans += max((ll)0, (ll) (a[i] - 1) * comp[i + 1] - comp[i]);
if(a[i] != 1){
join(i, i + 1);
}
e.pb({comp[i + 1] - comp[i], i});
}
sort(all(e));
for(int i = 0; i < sz(e); i++){
if(join(e[i][1], e[i][1] + 1)){
ans += e[i][0];
}
}
return ans;
}
Compilation message
railroad.cpp: In function 'long long int plan_roller_coaster(std::vector<int>, std::vector<int>)':
railroad.cpp:63:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
63 | for(int i = 0; i < sz(comp) - 1; i++){
| ~~^~~~~~~~~~~~~~
railroad.cpp:72:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::array<int, 2> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
72 | for(int i = 0; i < sz(e); i++){
| ^
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
212 KB |
n = 2 |
2 |
Correct |
1 ms |
212 KB |
n = 2 |
3 |
Correct |
1 ms |
212 KB |
n = 2 |
4 |
Correct |
1 ms |
300 KB |
n = 2 |
5 |
Correct |
1 ms |
296 KB |
n = 2 |
6 |
Correct |
1 ms |
212 KB |
n = 2 |
7 |
Correct |
1 ms |
212 KB |
n = 3 |
8 |
Correct |
1 ms |
300 KB |
n = 3 |
9 |
Correct |
1 ms |
212 KB |
n = 3 |
10 |
Correct |
1 ms |
212 KB |
n = 8 |
11 |
Correct |
1 ms |
212 KB |
n = 8 |
12 |
Incorrect |
1 ms |
212 KB |
answer is not correct: 20130219805 instead of 2726473632 |
13 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
212 KB |
n = 2 |
2 |
Correct |
1 ms |
212 KB |
n = 2 |
3 |
Correct |
1 ms |
212 KB |
n = 2 |
4 |
Correct |
1 ms |
300 KB |
n = 2 |
5 |
Correct |
1 ms |
296 KB |
n = 2 |
6 |
Correct |
1 ms |
212 KB |
n = 2 |
7 |
Correct |
1 ms |
212 KB |
n = 3 |
8 |
Correct |
1 ms |
300 KB |
n = 3 |
9 |
Correct |
1 ms |
212 KB |
n = 3 |
10 |
Correct |
1 ms |
212 KB |
n = 8 |
11 |
Correct |
1 ms |
212 KB |
n = 8 |
12 |
Incorrect |
1 ms |
212 KB |
answer is not correct: 20130219805 instead of 2726473632 |
13 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
272 ms |
13248 KB |
n = 199999 |
2 |
Correct |
290 ms |
17052 KB |
n = 199991 |
3 |
Incorrect |
264 ms |
17180 KB |
answer is not correct: -5114528881159554314 instead of 1 |
4 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
212 KB |
n = 2 |
2 |
Correct |
1 ms |
212 KB |
n = 2 |
3 |
Correct |
1 ms |
212 KB |
n = 2 |
4 |
Correct |
1 ms |
300 KB |
n = 2 |
5 |
Correct |
1 ms |
296 KB |
n = 2 |
6 |
Correct |
1 ms |
212 KB |
n = 2 |
7 |
Correct |
1 ms |
212 KB |
n = 3 |
8 |
Correct |
1 ms |
300 KB |
n = 3 |
9 |
Correct |
1 ms |
212 KB |
n = 3 |
10 |
Correct |
1 ms |
212 KB |
n = 8 |
11 |
Correct |
1 ms |
212 KB |
n = 8 |
12 |
Incorrect |
1 ms |
212 KB |
answer is not correct: 20130219805 instead of 2726473632 |
13 |
Halted |
0 ms |
0 KB |
- |