Submission #617145

# Submission time Handle Problem Language Result Execution time Memory
617145 2022-08-01T09:01:27 Z maximath_1 Roller Coaster Railroad (IOI16_railroad) C++11
100 / 100
173 ms 21084 KB
#include "railroad.h"
#include <bits/stdc++.h>
using namespace std;

#define ll long long
const int MX = 2e5 + 5;
const int inf = 1e9 + 69;

int par[MX];
int find(int nw){
	return par[nw] = (par[nw] == nw ? nw : find(par[nw]));
}
bool uni(int u, int v){
	u = find(u); v = find(v);
	par[u] = v; return (u != v);
}

ll plan_roller_coaster(vector<int> s, vector<int> t){
	int n = (int)s.size();
	for(int i = 0; i < MX; i ++) par[i] = i;

	vector<pair<int, pair<int, int> > > li, edges;
	for(int i = 0; i < n; i ++){ // edge si -> ti
		li.push_back({s[i], {1, i}});
		li.push_back({t[i], {-1, i}});
	} // edge inf -> 1
	li.push_back({inf, {1, n}});
	li.push_back({1, {-1, n}});

	sort(li.begin(), li.end());

	int balance = 0;
	ll ans = 0ll;
	for(int i = 0; i + 1 < li.size(); i ++){
		balance += li[i].second.first;
		ans += max(0, balance) * 1ll * (li[i + 1].first - li[i].first);
		edges.push_back({li[i + 1].first - li[i].first, {li[i + 1].second.second, li[i].second.second}});
		if(li[i + 1].first == li[i].first || balance != 0)
			uni(li[i + 1].second.second, li[i].second.second);
	}

	sort(edges.begin(), edges.end());
	for(auto i : edges) if(uni(i.second.first, i.second.second)) ans += i.first;

	return ans;
}

Compilation message

railroad.cpp: In function 'long long int plan_roller_coaster(std::vector<int>, std::vector<int>)':
railroad.cpp:34:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, std::pair<int, int> > >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   34 |  for(int i = 0; i + 1 < li.size(); i ++){
      |                 ~~~~~~^~~~~~~~~~~
# Verdict Execution time Memory Grader output
1 Correct 1 ms 980 KB n = 2
2 Correct 1 ms 980 KB n = 2
3 Correct 1 ms 980 KB n = 2
4 Correct 1 ms 980 KB n = 2
5 Correct 1 ms 1060 KB n = 2
6 Correct 1 ms 980 KB n = 2
7 Correct 1 ms 980 KB n = 3
8 Correct 1 ms 980 KB n = 3
9 Correct 1 ms 980 KB n = 3
10 Correct 1 ms 980 KB n = 8
11 Correct 1 ms 980 KB n = 8
12 Correct 2 ms 980 KB n = 8
13 Correct 1 ms 980 KB n = 8
14 Correct 1 ms 1008 KB n = 8
15 Correct 1 ms 980 KB n = 8
16 Correct 1 ms 980 KB n = 8
17 Correct 1 ms 980 KB n = 8
18 Correct 1 ms 1080 KB n = 8
19 Correct 1 ms 980 KB n = 3
20 Correct 1 ms 1084 KB n = 7
21 Correct 1 ms 1076 KB n = 8
22 Correct 1 ms 1080 KB n = 8
23 Correct 1 ms 980 KB n = 8
24 Correct 1 ms 980 KB n = 8
25 Correct 1 ms 980 KB n = 8
# Verdict Execution time Memory Grader output
1 Correct 1 ms 980 KB n = 2
2 Correct 1 ms 980 KB n = 2
3 Correct 1 ms 980 KB n = 2
4 Correct 1 ms 980 KB n = 2
5 Correct 1 ms 1060 KB n = 2
6 Correct 1 ms 980 KB n = 2
7 Correct 1 ms 980 KB n = 3
8 Correct 1 ms 980 KB n = 3
9 Correct 1 ms 980 KB n = 3
10 Correct 1 ms 980 KB n = 8
11 Correct 1 ms 980 KB n = 8
12 Correct 2 ms 980 KB n = 8
13 Correct 1 ms 980 KB n = 8
14 Correct 1 ms 1008 KB n = 8
15 Correct 1 ms 980 KB n = 8
16 Correct 1 ms 980 KB n = 8
17 Correct 1 ms 980 KB n = 8
18 Correct 1 ms 1080 KB n = 8
19 Correct 1 ms 980 KB n = 3
20 Correct 1 ms 1084 KB n = 7
21 Correct 1 ms 1076 KB n = 8
22 Correct 1 ms 1080 KB n = 8
23 Correct 1 ms 980 KB n = 8
24 Correct 1 ms 980 KB n = 8
25 Correct 1 ms 980 KB n = 8
26 Correct 2 ms 1080 KB n = 8
27 Correct 1 ms 980 KB n = 8
28 Correct 2 ms 980 KB n = 8
29 Correct 1 ms 980 KB n = 16
30 Correct 1 ms 1084 KB n = 16
31 Correct 1 ms 1108 KB n = 16
32 Correct 1 ms 1084 KB n = 16
33 Correct 1 ms 980 KB n = 16
34 Correct 2 ms 1076 KB n = 16
35 Correct 1 ms 980 KB n = 16
36 Correct 1 ms 980 KB n = 15
37 Correct 1 ms 980 KB n = 8
38 Correct 1 ms 980 KB n = 16
39 Correct 1 ms 1076 KB n = 16
40 Correct 1 ms 980 KB n = 9
41 Correct 1 ms 980 KB n = 16
42 Correct 1 ms 1080 KB n = 16
43 Correct 1 ms 1080 KB n = 16
44 Correct 1 ms 980 KB n = 9
45 Correct 1 ms 1084 KB n = 15
46 Correct 1 ms 980 KB n = 16
47 Correct 1 ms 1076 KB n = 16
48 Correct 1 ms 1024 KB n = 16
# Verdict Execution time Memory Grader output
1 Correct 145 ms 18140 KB n = 199999
2 Correct 163 ms 18144 KB n = 199991
3 Correct 150 ms 18144 KB n = 199993
4 Correct 107 ms 16264 KB n = 152076
5 Correct 63 ms 9356 KB n = 93249
6 Correct 153 ms 18132 KB n = 199910
7 Correct 137 ms 18100 KB n = 199999
8 Correct 139 ms 18140 KB n = 199997
9 Correct 131 ms 16972 KB n = 171294
10 Correct 103 ms 15728 KB n = 140872
11 Correct 147 ms 18080 KB n = 199886
12 Correct 165 ms 18100 KB n = 199996
13 Correct 148 ms 18096 KB n = 200000
14 Correct 155 ms 18136 KB n = 199998
15 Correct 164 ms 18100 KB n = 200000
16 Correct 160 ms 18096 KB n = 199998
17 Correct 144 ms 18096 KB n = 200000
18 Correct 137 ms 17668 KB n = 190000
19 Correct 121 ms 17284 KB n = 177777
20 Correct 82 ms 9516 KB n = 100000
21 Correct 146 ms 18096 KB n = 200000
22 Correct 140 ms 18112 KB n = 200000
23 Correct 137 ms 18108 KB n = 200000
# Verdict Execution time Memory Grader output
1 Correct 1 ms 980 KB n = 2
2 Correct 1 ms 980 KB n = 2
3 Correct 1 ms 980 KB n = 2
4 Correct 1 ms 980 KB n = 2
5 Correct 1 ms 1060 KB n = 2
6 Correct 1 ms 980 KB n = 2
7 Correct 1 ms 980 KB n = 3
8 Correct 1 ms 980 KB n = 3
9 Correct 1 ms 980 KB n = 3
10 Correct 1 ms 980 KB n = 8
11 Correct 1 ms 980 KB n = 8
12 Correct 2 ms 980 KB n = 8
13 Correct 1 ms 980 KB n = 8
14 Correct 1 ms 1008 KB n = 8
15 Correct 1 ms 980 KB n = 8
16 Correct 1 ms 980 KB n = 8
17 Correct 1 ms 980 KB n = 8
18 Correct 1 ms 1080 KB n = 8
19 Correct 1 ms 980 KB n = 3
20 Correct 1 ms 1084 KB n = 7
21 Correct 1 ms 1076 KB n = 8
22 Correct 1 ms 1080 KB n = 8
23 Correct 1 ms 980 KB n = 8
24 Correct 1 ms 980 KB n = 8
25 Correct 1 ms 980 KB n = 8
26 Correct 2 ms 1080 KB n = 8
27 Correct 1 ms 980 KB n = 8
28 Correct 2 ms 980 KB n = 8
29 Correct 1 ms 980 KB n = 16
30 Correct 1 ms 1084 KB n = 16
31 Correct 1 ms 1108 KB n = 16
32 Correct 1 ms 1084 KB n = 16
33 Correct 1 ms 980 KB n = 16
34 Correct 2 ms 1076 KB n = 16
35 Correct 1 ms 980 KB n = 16
36 Correct 1 ms 980 KB n = 15
37 Correct 1 ms 980 KB n = 8
38 Correct 1 ms 980 KB n = 16
39 Correct 1 ms 1076 KB n = 16
40 Correct 1 ms 980 KB n = 9
41 Correct 1 ms 980 KB n = 16
42 Correct 1 ms 1080 KB n = 16
43 Correct 1 ms 1080 KB n = 16
44 Correct 1 ms 980 KB n = 9
45 Correct 1 ms 1084 KB n = 15
46 Correct 1 ms 980 KB n = 16
47 Correct 1 ms 1076 KB n = 16
48 Correct 1 ms 1024 KB n = 16
49 Correct 145 ms 18140 KB n = 199999
50 Correct 163 ms 18144 KB n = 199991
51 Correct 150 ms 18144 KB n = 199993
52 Correct 107 ms 16264 KB n = 152076
53 Correct 63 ms 9356 KB n = 93249
54 Correct 153 ms 18132 KB n = 199910
55 Correct 137 ms 18100 KB n = 199999
56 Correct 139 ms 18140 KB n = 199997
57 Correct 131 ms 16972 KB n = 171294
58 Correct 103 ms 15728 KB n = 140872
59 Correct 147 ms 18080 KB n = 199886
60 Correct 165 ms 18100 KB n = 199996
61 Correct 148 ms 18096 KB n = 200000
62 Correct 155 ms 18136 KB n = 199998
63 Correct 164 ms 18100 KB n = 200000
64 Correct 160 ms 18096 KB n = 199998
65 Correct 144 ms 18096 KB n = 200000
66 Correct 137 ms 17668 KB n = 190000
67 Correct 121 ms 17284 KB n = 177777
68 Correct 82 ms 9516 KB n = 100000
69 Correct 146 ms 18096 KB n = 200000
70 Correct 140 ms 18112 KB n = 200000
71 Correct 137 ms 18108 KB n = 200000
72 Correct 173 ms 20740 KB n = 200000
73 Correct 151 ms 20828 KB n = 200000
74 Correct 146 ms 20956 KB n = 200000
75 Correct 158 ms 20904 KB n = 200000
76 Correct 142 ms 20868 KB n = 200000
77 Correct 159 ms 20912 KB n = 200000
78 Correct 146 ms 21048 KB n = 200000
79 Correct 145 ms 20272 KB n = 184307
80 Correct 56 ms 10140 KB n = 76040
81 Correct 168 ms 20852 KB n = 199981
82 Correct 158 ms 20992 KB n = 199994
83 Correct 152 ms 20948 KB n = 199996
84 Correct 151 ms 20868 KB n = 199998
85 Correct 170 ms 21016 KB n = 200000
86 Correct 160 ms 20812 KB n = 199998
87 Correct 145 ms 20964 KB n = 200000
88 Correct 147 ms 20920 KB n = 200000
89 Correct 146 ms 21040 KB n = 200000
90 Correct 155 ms 20916 KB n = 200000
91 Correct 153 ms 21084 KB n = 200000
92 Correct 144 ms 20948 KB n = 200000