#include "railroad.h"
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
typedef tuple<int,int,int> tu;
ll ans,cross[400010],dis[400010];
int p[400010],lv[400010];
vector<int> adj[400010];
vector<tu> v;
set<int> ss;
map<int,int> c;
void dfs(int x, int val) {
p[x]=val;
for (auto s : adj[x]) {
if (p[s]==-1) dfs(s,val);
}
}
int find(int x) {
while (x!=p[x]) x=p[x];
return x;
}
bool same(int x, int y) {
return find(x)==find(y);
}
void unite(int x, int y) {
x=find(x); y=find(y);
if (lv[x]>lv[y]) p[y]=x;
else if (lv[y]>lv[x]) p[x]=y;
else {
p[x]=y; ++lv[y];
}
}
long long plan_roller_coaster(std::vector<int> s, std::vector<int> t) {
int n = (int) s.size();
memset(p,-1,sizeof(p));
for (int i=0; i<n; ++i) {
ss.insert(s[i]);
ss.insert(t[i]);
}
ss.insert(1000000010);
int cnt=0,pre=0;
for (auto it=ss.begin(); it!=ss.end(); ++it) {
if (cnt) v.push_back(tu(*it-pre,cnt,cnt-1)), dis[cnt]=*it-pre;
c[*it]=cnt++;
pre=*it;
}
--cnt;
for (int i=0; i<n; ++i) {
int st=c[s[i]], to=c[t[i]];
cross[st+1]++;
cross[to+1]--;
adj[st].push_back(to);
}
adj[cnt].push_back(0);
--cross[1];
for (int i=1; i<=cnt; ++i) {
cross[i]+=cross[i-1];
if (cross[i]>0) {
adj[i].push_back(i-1);
ans+=dis[i]*cross[i];
} else if (cross[i]<0) {
adj[i-1].push_back(i);
}
}
for (int i=0; i<=cnt; ++i) {
if (p[i]==-1) dfs(i,i), lv[i]=1;
}
sort(v.begin(),v.end());
for (auto s : v) {
int a,b,w;
tie(w,a,b)=s;
if (!same(a,b)) {
unite(a,b);
ans+=w;
}
}
return ans;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
6 ms |
11220 KB |
n = 2 |
2 |
Correct |
6 ms |
11252 KB |
n = 2 |
3 |
Correct |
6 ms |
11260 KB |
n = 2 |
4 |
Correct |
6 ms |
11220 KB |
n = 2 |
5 |
Correct |
6 ms |
11220 KB |
n = 2 |
6 |
Correct |
6 ms |
11220 KB |
n = 2 |
7 |
Correct |
5 ms |
11220 KB |
n = 3 |
8 |
Correct |
6 ms |
11220 KB |
n = 3 |
9 |
Correct |
6 ms |
11220 KB |
n = 3 |
10 |
Correct |
5 ms |
11220 KB |
n = 8 |
11 |
Correct |
6 ms |
11220 KB |
n = 8 |
12 |
Correct |
6 ms |
11240 KB |
n = 8 |
13 |
Correct |
6 ms |
11220 KB |
n = 8 |
14 |
Correct |
6 ms |
11220 KB |
n = 8 |
15 |
Correct |
6 ms |
11220 KB |
n = 8 |
16 |
Correct |
6 ms |
11220 KB |
n = 8 |
17 |
Correct |
6 ms |
11220 KB |
n = 8 |
18 |
Correct |
6 ms |
11220 KB |
n = 8 |
19 |
Correct |
6 ms |
11272 KB |
n = 3 |
20 |
Correct |
6 ms |
11220 KB |
n = 7 |
21 |
Correct |
6 ms |
11220 KB |
n = 8 |
22 |
Correct |
5 ms |
11176 KB |
n = 8 |
23 |
Correct |
6 ms |
11220 KB |
n = 8 |
24 |
Correct |
6 ms |
11220 KB |
n = 8 |
25 |
Correct |
7 ms |
11220 KB |
n = 8 |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
6 ms |
11220 KB |
n = 2 |
2 |
Correct |
6 ms |
11252 KB |
n = 2 |
3 |
Correct |
6 ms |
11260 KB |
n = 2 |
4 |
Correct |
6 ms |
11220 KB |
n = 2 |
5 |
Correct |
6 ms |
11220 KB |
n = 2 |
6 |
Correct |
6 ms |
11220 KB |
n = 2 |
7 |
Correct |
5 ms |
11220 KB |
n = 3 |
8 |
Correct |
6 ms |
11220 KB |
n = 3 |
9 |
Correct |
6 ms |
11220 KB |
n = 3 |
10 |
Correct |
5 ms |
11220 KB |
n = 8 |
11 |
Correct |
6 ms |
11220 KB |
n = 8 |
12 |
Correct |
6 ms |
11240 KB |
n = 8 |
13 |
Correct |
6 ms |
11220 KB |
n = 8 |
14 |
Correct |
6 ms |
11220 KB |
n = 8 |
15 |
Correct |
6 ms |
11220 KB |
n = 8 |
16 |
Correct |
6 ms |
11220 KB |
n = 8 |
17 |
Correct |
6 ms |
11220 KB |
n = 8 |
18 |
Correct |
6 ms |
11220 KB |
n = 8 |
19 |
Correct |
6 ms |
11272 KB |
n = 3 |
20 |
Correct |
6 ms |
11220 KB |
n = 7 |
21 |
Correct |
6 ms |
11220 KB |
n = 8 |
22 |
Correct |
5 ms |
11176 KB |
n = 8 |
23 |
Correct |
6 ms |
11220 KB |
n = 8 |
24 |
Correct |
6 ms |
11220 KB |
n = 8 |
25 |
Correct |
7 ms |
11220 KB |
n = 8 |
26 |
Correct |
6 ms |
11220 KB |
n = 8 |
27 |
Correct |
5 ms |
11220 KB |
n = 8 |
28 |
Correct |
6 ms |
11260 KB |
n = 8 |
29 |
Correct |
6 ms |
11220 KB |
n = 16 |
30 |
Correct |
6 ms |
11220 KB |
n = 16 |
31 |
Correct |
5 ms |
11220 KB |
n = 16 |
32 |
Correct |
5 ms |
11220 KB |
n = 16 |
33 |
Correct |
6 ms |
11196 KB |
n = 16 |
34 |
Correct |
5 ms |
11248 KB |
n = 16 |
35 |
Correct |
5 ms |
11220 KB |
n = 16 |
36 |
Correct |
6 ms |
11220 KB |
n = 15 |
37 |
Correct |
6 ms |
11220 KB |
n = 8 |
38 |
Correct |
7 ms |
11220 KB |
n = 16 |
39 |
Correct |
6 ms |
11220 KB |
n = 16 |
40 |
Correct |
6 ms |
11220 KB |
n = 9 |
41 |
Correct |
6 ms |
11220 KB |
n = 16 |
42 |
Correct |
7 ms |
11220 KB |
n = 16 |
43 |
Correct |
6 ms |
11220 KB |
n = 16 |
44 |
Correct |
6 ms |
11220 KB |
n = 9 |
45 |
Correct |
6 ms |
11220 KB |
n = 15 |
46 |
Correct |
6 ms |
11220 KB |
n = 16 |
47 |
Correct |
6 ms |
11272 KB |
n = 16 |
48 |
Correct |
5 ms |
11292 KB |
n = 16 |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
722 ms |
83508 KB |
n = 199999 |
2 |
Correct |
741 ms |
78916 KB |
n = 199991 |
3 |
Correct |
708 ms |
83256 KB |
n = 199993 |
4 |
Correct |
506 ms |
62444 KB |
n = 152076 |
5 |
Correct |
273 ms |
42728 KB |
n = 93249 |
6 |
Correct |
598 ms |
66356 KB |
n = 199910 |
7 |
Correct |
671 ms |
77952 KB |
n = 199999 |
8 |
Correct |
562 ms |
68000 KB |
n = 199997 |
9 |
Correct |
604 ms |
69056 KB |
n = 171294 |
10 |
Correct |
475 ms |
58660 KB |
n = 140872 |
11 |
Correct |
600 ms |
67184 KB |
n = 199886 |
12 |
Correct |
695 ms |
79264 KB |
n = 199996 |
13 |
Correct |
568 ms |
71412 KB |
n = 200000 |
14 |
Correct |
575 ms |
81208 KB |
n = 199998 |
15 |
Correct |
581 ms |
79048 KB |
n = 200000 |
16 |
Correct |
575 ms |
80812 KB |
n = 199998 |
17 |
Correct |
626 ms |
79464 KB |
n = 200000 |
18 |
Correct |
614 ms |
76376 KB |
n = 190000 |
19 |
Correct |
546 ms |
75492 KB |
n = 177777 |
20 |
Correct |
257 ms |
45160 KB |
n = 100000 |
21 |
Correct |
660 ms |
78892 KB |
n = 200000 |
22 |
Correct |
644 ms |
76980 KB |
n = 200000 |
23 |
Correct |
726 ms |
83624 KB |
n = 200000 |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
6 ms |
11220 KB |
n = 2 |
2 |
Correct |
6 ms |
11252 KB |
n = 2 |
3 |
Correct |
6 ms |
11260 KB |
n = 2 |
4 |
Correct |
6 ms |
11220 KB |
n = 2 |
5 |
Correct |
6 ms |
11220 KB |
n = 2 |
6 |
Correct |
6 ms |
11220 KB |
n = 2 |
7 |
Correct |
5 ms |
11220 KB |
n = 3 |
8 |
Correct |
6 ms |
11220 KB |
n = 3 |
9 |
Correct |
6 ms |
11220 KB |
n = 3 |
10 |
Correct |
5 ms |
11220 KB |
n = 8 |
11 |
Correct |
6 ms |
11220 KB |
n = 8 |
12 |
Correct |
6 ms |
11240 KB |
n = 8 |
13 |
Correct |
6 ms |
11220 KB |
n = 8 |
14 |
Correct |
6 ms |
11220 KB |
n = 8 |
15 |
Correct |
6 ms |
11220 KB |
n = 8 |
16 |
Correct |
6 ms |
11220 KB |
n = 8 |
17 |
Correct |
6 ms |
11220 KB |
n = 8 |
18 |
Correct |
6 ms |
11220 KB |
n = 8 |
19 |
Correct |
6 ms |
11272 KB |
n = 3 |
20 |
Correct |
6 ms |
11220 KB |
n = 7 |
21 |
Correct |
6 ms |
11220 KB |
n = 8 |
22 |
Correct |
5 ms |
11176 KB |
n = 8 |
23 |
Correct |
6 ms |
11220 KB |
n = 8 |
24 |
Correct |
6 ms |
11220 KB |
n = 8 |
25 |
Correct |
7 ms |
11220 KB |
n = 8 |
26 |
Correct |
6 ms |
11220 KB |
n = 8 |
27 |
Correct |
5 ms |
11220 KB |
n = 8 |
28 |
Correct |
6 ms |
11260 KB |
n = 8 |
29 |
Correct |
6 ms |
11220 KB |
n = 16 |
30 |
Correct |
6 ms |
11220 KB |
n = 16 |
31 |
Correct |
5 ms |
11220 KB |
n = 16 |
32 |
Correct |
5 ms |
11220 KB |
n = 16 |
33 |
Correct |
6 ms |
11196 KB |
n = 16 |
34 |
Correct |
5 ms |
11248 KB |
n = 16 |
35 |
Correct |
5 ms |
11220 KB |
n = 16 |
36 |
Correct |
6 ms |
11220 KB |
n = 15 |
37 |
Correct |
6 ms |
11220 KB |
n = 8 |
38 |
Correct |
7 ms |
11220 KB |
n = 16 |
39 |
Correct |
6 ms |
11220 KB |
n = 16 |
40 |
Correct |
6 ms |
11220 KB |
n = 9 |
41 |
Correct |
6 ms |
11220 KB |
n = 16 |
42 |
Correct |
7 ms |
11220 KB |
n = 16 |
43 |
Correct |
6 ms |
11220 KB |
n = 16 |
44 |
Correct |
6 ms |
11220 KB |
n = 9 |
45 |
Correct |
6 ms |
11220 KB |
n = 15 |
46 |
Correct |
6 ms |
11220 KB |
n = 16 |
47 |
Correct |
6 ms |
11272 KB |
n = 16 |
48 |
Correct |
5 ms |
11292 KB |
n = 16 |
49 |
Correct |
722 ms |
83508 KB |
n = 199999 |
50 |
Correct |
741 ms |
78916 KB |
n = 199991 |
51 |
Correct |
708 ms |
83256 KB |
n = 199993 |
52 |
Correct |
506 ms |
62444 KB |
n = 152076 |
53 |
Correct |
273 ms |
42728 KB |
n = 93249 |
54 |
Correct |
598 ms |
66356 KB |
n = 199910 |
55 |
Correct |
671 ms |
77952 KB |
n = 199999 |
56 |
Correct |
562 ms |
68000 KB |
n = 199997 |
57 |
Correct |
604 ms |
69056 KB |
n = 171294 |
58 |
Correct |
475 ms |
58660 KB |
n = 140872 |
59 |
Correct |
600 ms |
67184 KB |
n = 199886 |
60 |
Correct |
695 ms |
79264 KB |
n = 199996 |
61 |
Correct |
568 ms |
71412 KB |
n = 200000 |
62 |
Correct |
575 ms |
81208 KB |
n = 199998 |
63 |
Correct |
581 ms |
79048 KB |
n = 200000 |
64 |
Correct |
575 ms |
80812 KB |
n = 199998 |
65 |
Correct |
626 ms |
79464 KB |
n = 200000 |
66 |
Correct |
614 ms |
76376 KB |
n = 190000 |
67 |
Correct |
546 ms |
75492 KB |
n = 177777 |
68 |
Correct |
257 ms |
45160 KB |
n = 100000 |
69 |
Correct |
660 ms |
78892 KB |
n = 200000 |
70 |
Correct |
644 ms |
76980 KB |
n = 200000 |
71 |
Correct |
726 ms |
83624 KB |
n = 200000 |
72 |
Correct |
736 ms |
83444 KB |
n = 200000 |
73 |
Correct |
751 ms |
78940 KB |
n = 200000 |
74 |
Correct |
724 ms |
82412 KB |
n = 200000 |
75 |
Correct |
652 ms |
87468 KB |
n = 200000 |
76 |
Correct |
650 ms |
87432 KB |
n = 200000 |
77 |
Correct |
366 ms |
52772 KB |
n = 200000 |
78 |
Correct |
387 ms |
49568 KB |
n = 200000 |
79 |
Correct |
667 ms |
75996 KB |
n = 184307 |
80 |
Correct |
219 ms |
38472 KB |
n = 76040 |
81 |
Correct |
635 ms |
70048 KB |
n = 199981 |
82 |
Correct |
603 ms |
82480 KB |
n = 199994 |
83 |
Correct |
589 ms |
74284 KB |
n = 199996 |
84 |
Correct |
615 ms |
84396 KB |
n = 199998 |
85 |
Correct |
566 ms |
82072 KB |
n = 200000 |
86 |
Correct |
631 ms |
84200 KB |
n = 199998 |
87 |
Correct |
699 ms |
83460 KB |
n = 200000 |
88 |
Correct |
709 ms |
83484 KB |
n = 200000 |
89 |
Correct |
679 ms |
87376 KB |
n = 200000 |
90 |
Correct |
662 ms |
82764 KB |
n = 200000 |
91 |
Correct |
698 ms |
80868 KB |
n = 200000 |
92 |
Correct |
746 ms |
87540 KB |
n = 200000 |