답안 #856020

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
856020 2023-10-02T14:34:53 Z qin Roller Coaster Railroad (IOI16_railroad) C++17
100 / 100
450 ms 53940 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){
				int b = a, c;
				while(a != rep[a]) a = rep[a];
				while(b != rep[b]) c = b, b = rep[b], rep[c] = a;
				return 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
# 결과 실행 시간 메모리 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 344 KB n = 3
8 Correct 0 ms 348 KB n = 3
9 Correct 0 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 348 KB n = 8
13 Correct 0 ms 348 KB n = 8
14 Correct 1 ms 348 KB n = 8
15 Correct 1 ms 344 KB n = 8
16 Correct 1 ms 600 KB n = 8
17 Correct 1 ms 348 KB n = 8
18 Correct 1 ms 500 KB n = 8
19 Correct 0 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 348 KB n = 8
25 Correct 0 ms 348 KB n = 8
# 결과 실행 시간 메모리 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 344 KB n = 3
8 Correct 0 ms 348 KB n = 3
9 Correct 0 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 348 KB n = 8
13 Correct 0 ms 348 KB n = 8
14 Correct 1 ms 348 KB n = 8
15 Correct 1 ms 344 KB n = 8
16 Correct 1 ms 600 KB n = 8
17 Correct 1 ms 348 KB n = 8
18 Correct 1 ms 500 KB n = 8
19 Correct 0 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 348 KB n = 8
25 Correct 0 ms 348 KB n = 8
26 Correct 0 ms 348 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 0 ms 348 KB n = 16
31 Correct 0 ms 348 KB n = 16
32 Correct 0 ms 348 KB n = 16
33 Correct 0 ms 348 KB n = 16
34 Correct 0 ms 348 KB n = 16
35 Correct 0 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 1 ms 348 KB n = 9
41 Correct 0 ms 348 KB n = 16
42 Correct 0 ms 344 KB n = 16
43 Correct 0 ms 348 KB n = 16
44 Correct 1 ms 344 KB n = 9
45 Correct 0 ms 348 KB n = 15
46 Correct 1 ms 348 KB n = 16
47 Correct 0 ms 348 KB n = 16
48 Correct 0 ms 348 KB n = 16
# 결과 실행 시간 메모리 Grader output
1 Correct 405 ms 46628 KB n = 199999
2 Correct 450 ms 46644 KB n = 199991
3 Correct 411 ms 46640 KB n = 199993
4 Correct 234 ms 33988 KB n = 152076
5 Correct 169 ms 22116 KB n = 93249
6 Correct 238 ms 36912 KB n = 199910
7 Correct 340 ms 45828 KB n = 199999
8 Correct 272 ms 37200 KB n = 199997
9 Correct 280 ms 37832 KB n = 171294
10 Correct 264 ms 31604 KB n = 140872
11 Correct 233 ms 37504 KB n = 199886
12 Correct 350 ms 46008 KB n = 199996
13 Correct 233 ms 38552 KB n = 200000
14 Correct 354 ms 47620 KB n = 199998
15 Correct 288 ms 46900 KB n = 200000
16 Correct 355 ms 48320 KB n = 199998
17 Correct 381 ms 46628 KB n = 200000
18 Correct 343 ms 44672 KB n = 190000
19 Correct 290 ms 42124 KB n = 177777
20 Correct 149 ms 23432 KB n = 100000
21 Correct 367 ms 46640 KB n = 200000
22 Correct 370 ms 53940 KB n = 200000
23 Correct 426 ms 46588 KB n = 200000
# 결과 실행 시간 메모리 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 344 KB n = 3
8 Correct 0 ms 348 KB n = 3
9 Correct 0 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 348 KB n = 8
13 Correct 0 ms 348 KB n = 8
14 Correct 1 ms 348 KB n = 8
15 Correct 1 ms 344 KB n = 8
16 Correct 1 ms 600 KB n = 8
17 Correct 1 ms 348 KB n = 8
18 Correct 1 ms 500 KB n = 8
19 Correct 0 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 348 KB n = 8
25 Correct 0 ms 348 KB n = 8
26 Correct 0 ms 348 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 0 ms 348 KB n = 16
31 Correct 0 ms 348 KB n = 16
32 Correct 0 ms 348 KB n = 16
33 Correct 0 ms 348 KB n = 16
34 Correct 0 ms 348 KB n = 16
35 Correct 0 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 1 ms 348 KB n = 9
41 Correct 0 ms 348 KB n = 16
42 Correct 0 ms 344 KB n = 16
43 Correct 0 ms 348 KB n = 16
44 Correct 1 ms 344 KB n = 9
45 Correct 0 ms 348 KB n = 15
46 Correct 1 ms 348 KB n = 16
47 Correct 0 ms 348 KB n = 16
48 Correct 0 ms 348 KB n = 16
49 Correct 405 ms 46628 KB n = 199999
50 Correct 450 ms 46644 KB n = 199991
51 Correct 411 ms 46640 KB n = 199993
52 Correct 234 ms 33988 KB n = 152076
53 Correct 169 ms 22116 KB n = 93249
54 Correct 238 ms 36912 KB n = 199910
55 Correct 340 ms 45828 KB n = 199999
56 Correct 272 ms 37200 KB n = 199997
57 Correct 280 ms 37832 KB n = 171294
58 Correct 264 ms 31604 KB n = 140872
59 Correct 233 ms 37504 KB n = 199886
60 Correct 350 ms 46008 KB n = 199996
61 Correct 233 ms 38552 KB n = 200000
62 Correct 354 ms 47620 KB n = 199998
63 Correct 288 ms 46900 KB n = 200000
64 Correct 355 ms 48320 KB n = 199998
65 Correct 381 ms 46628 KB n = 200000
66 Correct 343 ms 44672 KB n = 190000
67 Correct 290 ms 42124 KB n = 177777
68 Correct 149 ms 23432 KB n = 100000
69 Correct 367 ms 46640 KB n = 200000
70 Correct 370 ms 53940 KB n = 200000
71 Correct 426 ms 46588 KB n = 200000
72 Correct 394 ms 46640 KB n = 200000
73 Correct 428 ms 46836 KB n = 200000
74 Correct 442 ms 46520 KB n = 200000
75 Correct 366 ms 46860 KB n = 200000
76 Correct 389 ms 46472 KB n = 200000
77 Correct 149 ms 25064 KB n = 200000
78 Correct 236 ms 31400 KB n = 200000
79 Correct 341 ms 43432 KB n = 184307
80 Correct 95 ms 17072 KB n = 76040
81 Correct 273 ms 37748 KB n = 199981
82 Correct 322 ms 46132 KB n = 199994
83 Correct 246 ms 38632 KB n = 199996
84 Correct 282 ms 47644 KB n = 199998
85 Correct 289 ms 47020 KB n = 200000
86 Correct 306 ms 48276 KB n = 199998
87 Correct 382 ms 46872 KB n = 200000
88 Correct 439 ms 46516 KB n = 200000
89 Correct 412 ms 46856 KB n = 200000
90 Correct 337 ms 46468 KB n = 200000
91 Correct 402 ms 53920 KB n = 200000
92 Correct 425 ms 46540 KB n = 200000