oj.uz

Submission #856071

# Submission time Handle Problem Language Result Execution time Memory
856071 2023-10-02T16:17:39 Z qin Roller Coaster Railroad (IOI16_railroad) C++17
100 / 100
 465 ms 57276 KB 
#include <bits/stdc++.h>
#define fi first
#define se second
#define ssize(x) int(x.size())
#define pn printf("\n");
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<int, ll> pil;
typedef pair<ll, int> pli;
typedef pair<ll, ll> pll;
int inf = 1e09; ll infll = 1e18;
struct DSU{
		vector<int> rep, sz;
		void init(int n){
				rep.resize(n+1), sz.resize(n+1, 1);
				for(int i = 1; i <= n; ++i) rep[i] = i;
		} int Find(int a){
				return (a != rep[a]) ? rep[a] = Find(rep[a]) : a;
		} void Union(int a, int b){
				a = Find(a), b = Find(b);
				if(a == b) return;
				if(sz[a] < sz[b]) swap(a, b);
				rep[b] = a, sz[a] += sz[b];
		}
} DSU;
struct edge{
		int a, b; ll c;
		edge(){}
		edge(int A, int B, ll C) : a(A), b(B), c(C) {}
		bool operator<(const edge &x) const{ return c < x.c; }
};
ll plan_roller_coaster(vector<int> s, vector<int> e){
		s.emplace_back(inf+1), e.emplace_back(1); // dodatkowa stacja
		int m = ssize(s), a, b; // m - ilosc stacji (poczatkowych kr. w grafie, n - ilosc wierzcholkow
		
		set<int> sc; unordered_map<int, int> mp; int n = 0; // skalowanie
		for(int i = 0; i < m; ++i) sc.emplace(s[i]), sc.emplace(e[i]);
		for(int i : sc) mp[i] = ++n;
		vector<int> def_val(n+1);
		for(int i = 0; i < m; ++i) a = mp[s[i]], b = mp[e[i]],
								   def_val[a] = s[i], s[i] = a, def_val[b] = e[i], e[i] = b;
		
		DSU.init(n);
		ll result = 0;
		vector<int> pf_sum(n+1); // trzymamy w tym roznice miedzy krawedziami
		for(int i = 0; i < m; ++i) ++pf_sum[s[i]], --pf_sum[e[i]], DSU.Union(s[i], e[i]);
		for(int i = 1; i < n; ++i){
				pf_sum[i] += pf_sum[i-1];
				if(pf_sum[i] > 0) result += pf_sum[i] * ll(def_val[i+1]-def_val[i]); 
				if(pf_sum[i]) DSU.Union(i, i+1);
		}
		
		vector<edge> MST; //MST
		for(int i = 1; i < n; ++i) if(DSU.Find(i) != DSU.Find(i+1)) MST.emplace_back(i, i+1, def_val[i+1]-def_val[i]);
		sort(MST.begin(), MST.end());
		for(edge i : MST) if(DSU.Find(i.a) != DSU.Find(i.b)) DSU.Union(i.a, i.b), result += i.c;
		
		return result;
}
#ifdef LOCAL
int main(){
		int n; scanf("%d", &n);
		vector<int> s(n), e(n);
		for(int i = 0; i < n; ++i) scanf("%d%d", &s[i], &e[i]);
		printf("%lld\n", plan_roller_coaster(s, e));
		return 0;
}
#endif

Subtask #1
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB n = 2
2 Correct 0 ms 348 KB n = 2
3 Correct 0 ms 348 KB n = 2
4 Correct 0 ms 348 KB n = 2
5 Correct 0 ms 348 KB n = 2
6 Correct 0 ms 348 KB n = 2
7 Correct 0 ms 348 KB n = 3
8 Correct 1 ms 348 KB n = 3
9 Correct 1 ms 348 KB n = 3
10 Correct 0 ms 348 KB n = 8
11 Correct 0 ms 348 KB n = 8
12 Correct 0 ms 344 KB n = 8
13 Correct 0 ms 348 KB n = 8
14 Correct 0 ms 344 KB n = 8
15 Correct 0 ms 348 KB n = 8
16 Correct 0 ms 348 KB n = 8
17 Correct 0 ms 348 KB n = 8
18 Correct 1 ms 348 KB n = 8
19 Correct 1 ms 348 KB n = 3
20 Correct 0 ms 348 KB n = 7
21 Correct 0 ms 348 KB n = 8
22 Correct 0 ms 348 KB n = 8
23 Correct 0 ms 348 KB n = 8
24 Correct 0 ms 344 KB n = 8
25 Correct 0 ms 348 KB n = 8

Subtask #2
23.0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB n = 2
2 Correct 0 ms 348 KB n = 2
3 Correct 0 ms 348 KB n = 2
4 Correct 0 ms 348 KB n = 2
5 Correct 0 ms 348 KB n = 2
6 Correct 0 ms 348 KB n = 2
7 Correct 0 ms 348 KB n = 3
8 Correct 1 ms 348 KB n = 3
9 Correct 1 ms 348 KB n = 3
10 Correct 0 ms 348 KB n = 8
11 Correct 0 ms 348 KB n = 8
12 Correct 0 ms 344 KB n = 8
13 Correct 0 ms 348 KB n = 8
14 Correct 0 ms 344 KB n = 8
15 Correct 0 ms 348 KB n = 8
16 Correct 0 ms 348 KB n = 8
17 Correct 0 ms 348 KB n = 8
18 Correct 1 ms 348 KB n = 8
19 Correct 1 ms 348 KB n = 3
20 Correct 0 ms 348 KB n = 7
21 Correct 0 ms 348 KB n = 8
22 Correct 0 ms 348 KB n = 8
23 Correct 0 ms 348 KB n = 8
24 Correct 0 ms 344 KB n = 8
25 Correct 0 ms 348 KB n = 8
26 Correct 0 ms 600 KB n = 8
27 Correct 0 ms 348 KB n = 8
28 Correct 0 ms 348 KB n = 8
29 Correct 0 ms 348 KB n = 16
30 Correct 1 ms 344 KB n = 16
31 Correct 0 ms 344 KB n = 16
32 Correct 0 ms 348 KB n = 16
33 Correct 0 ms 348 KB n = 16
34 Correct 1 ms 348 KB n = 16
35 Correct 1 ms 348 KB n = 16
36 Correct 0 ms 348 KB n = 15
37 Correct 0 ms 348 KB n = 8
38 Correct 0 ms 348 KB n = 16
39 Correct 0 ms 348 KB n = 16
40 Correct 0 ms 344 KB n = 9
41 Correct 0 ms 348 KB n = 16
42 Correct 0 ms 348 KB n = 16
43 Correct 0 ms 348 KB n = 16
44 Correct 0 ms 348 KB n = 9
45 Correct 0 ms 344 KB n = 15
46 Correct 0 ms 348 KB n = 16
47 Correct 0 ms 348 KB n = 16
48 Correct 0 ms 348 KB n = 16

Subtask #3
30.0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 435 ms 50004 KB n = 199999
2 Correct 465 ms 50068 KB n = 199991
3 Correct 450 ms 49820 KB n = 199993
4 Correct 330 ms 36144 KB n = 152076
5 Correct 142 ms 23476 KB n = 93249
6 Correct 290 ms 39236 KB n = 199910
7 Correct 342 ms 48580 KB n = 199999
8 Correct 297 ms 39552 KB n = 199997
9 Correct 414 ms 40804 KB n = 171294
10 Correct 258 ms 34080 KB n = 140872
11 Correct 319 ms 39664 KB n = 199886
12 Correct 360 ms 48932 KB n = 199996
13 Correct 249 ms 41124 KB n = 200000
14 Correct 294 ms 50276 KB n = 199998
15 Correct 259 ms 49516 KB n = 200000
16 Correct 343 ms 51356 KB n = 199998
17 Correct 326 ms 49868 KB n = 200000
18 Correct 354 ms 47780 KB n = 190000
19 Correct 287 ms 45224 KB n = 177777
20 Correct 113 ms 25100 KB n = 100000
21 Correct 331 ms 49960 KB n = 200000
22 Correct 369 ms 57276 KB n = 200000
23 Correct 367 ms 49812 KB n = 200000

Subtask #4
36.0 / 36.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB n = 2
2 Correct 0 ms 348 KB n = 2
3 Correct 0 ms 348 KB n = 2
4 Correct 0 ms 348 KB n = 2
5 Correct 0 ms 348 KB n = 2
6 Correct 0 ms 348 KB n = 2
7 Correct 0 ms 348 KB n = 3
8 Correct 1 ms 348 KB n = 3
9 Correct 1 ms 348 KB n = 3
10 Correct 0 ms 348 KB n = 8
11 Correct 0 ms 348 KB n = 8
12 Correct 0 ms 344 KB n = 8
13 Correct 0 ms 348 KB n = 8
14 Correct 0 ms 344 KB n = 8
15 Correct 0 ms 348 KB n = 8
16 Correct 0 ms 348 KB n = 8
17 Correct 0 ms 348 KB n = 8
18 Correct 1 ms 348 KB n = 8
19 Correct 1 ms 348 KB n = 3
20 Correct 0 ms 348 KB n = 7
21 Correct 0 ms 348 KB n = 8
22 Correct 0 ms 348 KB n = 8
23 Correct 0 ms 348 KB n = 8
24 Correct 0 ms 344 KB n = 8
25 Correct 0 ms 348 KB n = 8
26 Correct 0 ms 600 KB n = 8
27 Correct 0 ms 348 KB n = 8
28 Correct 0 ms 348 KB n = 8
29 Correct 0 ms 348 KB n = 16
30 Correct 1 ms 344 KB n = 16
31 Correct 0 ms 344 KB n = 16
32 Correct 0 ms 348 KB n = 16
33 Correct 0 ms 348 KB n = 16
34 Correct 1 ms 348 KB n = 16
35 Correct 1 ms 348 KB n = 16
36 Correct 0 ms 348 KB n = 15
37 Correct 0 ms 348 KB n = 8
38 Correct 0 ms 348 KB n = 16
39 Correct 0 ms 348 KB n = 16
40 Correct 0 ms 344 KB n = 9
41 Correct 0 ms 348 KB n = 16
42 Correct 0 ms 348 KB n = 16
43 Correct 0 ms 348 KB n = 16
44 Correct 0 ms 348 KB n = 9
45 Correct 0 ms 344 KB n = 15
46 Correct 0 ms 348 KB n = 16
47 Correct 0 ms 348 KB n = 16
48 Correct 0 ms 348 KB n = 16
49 Correct 435 ms 50004 KB n = 199999
50 Correct 465 ms 50068 KB n = 199991
51 Correct 450 ms 49820 KB n = 199993
52 Correct 330 ms 36144 KB n = 152076
53 Correct 142 ms 23476 KB n = 93249
54 Correct 290 ms 39236 KB n = 199910
55 Correct 342 ms 48580 KB n = 199999
56 Correct 297 ms 39552 KB n = 199997
57 Correct 414 ms 40804 KB n = 171294
58 Correct 258 ms 34080 KB n = 140872
59 Correct 319 ms 39664 KB n = 199886
60 Correct 360 ms 48932 KB n = 199996
61 Correct 249 ms 41124 KB n = 200000
62 Correct 294 ms 50276 KB n = 199998
63 Correct 259 ms 49516 KB n = 200000
64 Correct 343 ms 51356 KB n = 199998
65 Correct 326 ms 49868 KB n = 200000
66 Correct 354 ms 47780 KB n = 190000
67 Correct 287 ms 45224 KB n = 177777
68 Correct 113 ms 25100 KB n = 100000
69 Correct 331 ms 49960 KB n = 200000
70 Correct 369 ms 57276 KB n = 200000
71 Correct 367 ms 49812 KB n = 200000
72 Correct 401 ms 49844 KB n = 200000
73 Correct 401 ms 49972 KB n = 200000
74 Correct 377 ms 50036 KB n = 200000
75 Correct 353 ms 49932 KB n = 200000
76 Correct 381 ms 49968 KB n = 200000
77 Correct 159 ms 28408 KB n = 200000
78 Correct 180 ms 35308 KB n = 200000
79 Correct 349 ms 46516 KB n = 184307
80 Correct 86 ms 18704 KB n = 76040
81 Correct 254 ms 39660 KB n = 199981
82 Correct 308 ms 48808 KB n = 199994
83 Correct 236 ms 41116 KB n = 199996
84 Correct 317 ms 50160 KB n = 199998
85 Correct 287 ms 49460 KB n = 200000
86 Correct 346 ms 51456 KB n = 199998
87 Correct 351 ms 50004 KB n = 200000
88 Correct 370 ms 49956 KB n = 200000
89 Correct 379 ms 50168 KB n = 200000
90 Correct 363 ms 50200 KB n = 200000
91 Correct 397 ms 57056 KB n = 200000
92 Correct 409 ms 50056 KB n = 200000