Submission #426368

# Submission time Handle Problem Language Result Execution time Memory
426368 2021-06-13T20:19:38 Z timmyfeng Roller Coaster Railroad (IOI16_railroad) C++17
100 / 100
1888 ms 59676 KB
#include <bits/stdc++.h>
using namespace std;

unordered_map<int, int> par;

int find(int u) {
    if (par.count(u) > 0) {
        return par[u] = find(par[u]);
    } else {
        return u;
    }
}

bool unite(int u, int v) {
    u = find(u), v = find(v);
    if (u != v) {
        par[u] = v;
        return true;
    }
    return false;
}
 
long long plan_roller_coaster(vector<int> s, vector<int> t) {
    s.push_back(INT_MAX), t.push_back(1);
    int n = s.size();

    map<int, int> events;
    for (int i = 0; i < n; ++i) {
        unite(s[i], t[i]);
        ++events[s[i]];
        --events[t[i]];
    }

    long long ans = 0;
    int prv = 1, balance = 0;
    vector<array<int, 3>> edges;
    for (auto [x, d] : events) {
        if (balance != 0) {
            ans += (long long) (x - prv) * max(0, balance);
            unite(prv, x);
        }

        edges.push_back({x - prv, prv, x});
        balance += d;
        prv = x;
    }
    sort(edges.begin(), edges.end());

    for (auto [w, u, v] : edges) {
        if (unite(u, v)) {
            ans += w;
        }
    }

    return ans;
}
# Verdict Execution time Memory Grader output
1 Correct 1 ms 204 KB n = 2
2 Correct 1 ms 204 KB n = 2
3 Correct 1 ms 204 KB n = 2
4 Correct 1 ms 204 KB n = 2
5 Correct 1 ms 204 KB n = 2
6 Correct 1 ms 204 KB n = 2
7 Correct 1 ms 204 KB n = 3
8 Correct 1 ms 204 KB n = 3
9 Correct 1 ms 204 KB n = 3
10 Correct 1 ms 204 KB n = 8
11 Correct 1 ms 204 KB n = 8
12 Correct 1 ms 204 KB n = 8
13 Correct 1 ms 204 KB n = 8
14 Correct 1 ms 204 KB n = 8
15 Correct 1 ms 204 KB n = 8
16 Correct 1 ms 204 KB n = 8
17 Correct 1 ms 204 KB n = 8
18 Correct 1 ms 204 KB n = 8
19 Correct 0 ms 204 KB n = 3
20 Correct 1 ms 204 KB n = 7
21 Correct 1 ms 204 KB n = 8
22 Correct 1 ms 204 KB n = 8
23 Correct 1 ms 204 KB n = 8
24 Correct 1 ms 204 KB n = 8
25 Correct 1 ms 204 KB n = 8
# Verdict Execution time Memory Grader output
1 Correct 1 ms 204 KB n = 2
2 Correct 1 ms 204 KB n = 2
3 Correct 1 ms 204 KB n = 2
4 Correct 1 ms 204 KB n = 2
5 Correct 1 ms 204 KB n = 2
6 Correct 1 ms 204 KB n = 2
7 Correct 1 ms 204 KB n = 3
8 Correct 1 ms 204 KB n = 3
9 Correct 1 ms 204 KB n = 3
10 Correct 1 ms 204 KB n = 8
11 Correct 1 ms 204 KB n = 8
12 Correct 1 ms 204 KB n = 8
13 Correct 1 ms 204 KB n = 8
14 Correct 1 ms 204 KB n = 8
15 Correct 1 ms 204 KB n = 8
16 Correct 1 ms 204 KB n = 8
17 Correct 1 ms 204 KB n = 8
18 Correct 1 ms 204 KB n = 8
19 Correct 0 ms 204 KB n = 3
20 Correct 1 ms 204 KB n = 7
21 Correct 1 ms 204 KB n = 8
22 Correct 1 ms 204 KB n = 8
23 Correct 1 ms 204 KB n = 8
24 Correct 1 ms 204 KB n = 8
25 Correct 1 ms 204 KB n = 8
26 Correct 1 ms 204 KB n = 8
27 Correct 1 ms 204 KB n = 8
28 Correct 1 ms 204 KB n = 8
29 Correct 1 ms 204 KB n = 16
30 Correct 1 ms 204 KB n = 16
31 Correct 1 ms 204 KB n = 16
32 Correct 1 ms 204 KB n = 16
33 Correct 1 ms 204 KB n = 16
34 Correct 1 ms 204 KB n = 16
35 Correct 1 ms 204 KB n = 16
36 Correct 1 ms 204 KB n = 15
37 Correct 0 ms 204 KB n = 8
38 Correct 1 ms 204 KB n = 16
39 Correct 1 ms 204 KB n = 16
40 Correct 1 ms 204 KB n = 9
41 Correct 1 ms 204 KB n = 16
42 Correct 1 ms 204 KB n = 16
43 Correct 1 ms 204 KB n = 16
44 Correct 1 ms 204 KB n = 9
45 Correct 1 ms 204 KB n = 15
46 Correct 1 ms 204 KB n = 16
47 Correct 0 ms 204 KB n = 16
48 Correct 1 ms 204 KB n = 16
# Verdict Execution time Memory Grader output
1 Correct 1712 ms 45776 KB n = 199999
2 Correct 1888 ms 49568 KB n = 199991
3 Correct 1567 ms 49544 KB n = 199993
4 Correct 1118 ms 38072 KB n = 152076
5 Correct 622 ms 23824 KB n = 93249
6 Correct 1090 ms 39804 KB n = 199910
7 Correct 1168 ms 48828 KB n = 199999
8 Correct 937 ms 40008 KB n = 199997
9 Correct 1449 ms 41416 KB n = 171294
10 Correct 1014 ms 36140 KB n = 140872
11 Correct 993 ms 40188 KB n = 199886
12 Correct 1326 ms 49560 KB n = 199996
13 Correct 925 ms 41860 KB n = 200000
14 Correct 1292 ms 53092 KB n = 199998
15 Correct 1022 ms 50344 KB n = 200000
16 Correct 1528 ms 51448 KB n = 199998
17 Correct 1343 ms 59672 KB n = 200000
18 Correct 1515 ms 49032 KB n = 190000
19 Correct 1235 ms 54476 KB n = 177777
20 Correct 511 ms 24600 KB n = 100000
21 Correct 1604 ms 48992 KB n = 200000
22 Correct 1205 ms 49992 KB n = 200000
23 Correct 1408 ms 49072 KB n = 200000
# Verdict Execution time Memory Grader output
1 Correct 1 ms 204 KB n = 2
2 Correct 1 ms 204 KB n = 2
3 Correct 1 ms 204 KB n = 2
4 Correct 1 ms 204 KB n = 2
5 Correct 1 ms 204 KB n = 2
6 Correct 1 ms 204 KB n = 2
7 Correct 1 ms 204 KB n = 3
8 Correct 1 ms 204 KB n = 3
9 Correct 1 ms 204 KB n = 3
10 Correct 1 ms 204 KB n = 8
11 Correct 1 ms 204 KB n = 8
12 Correct 1 ms 204 KB n = 8
13 Correct 1 ms 204 KB n = 8
14 Correct 1 ms 204 KB n = 8
15 Correct 1 ms 204 KB n = 8
16 Correct 1 ms 204 KB n = 8
17 Correct 1 ms 204 KB n = 8
18 Correct 1 ms 204 KB n = 8
19 Correct 0 ms 204 KB n = 3
20 Correct 1 ms 204 KB n = 7
21 Correct 1 ms 204 KB n = 8
22 Correct 1 ms 204 KB n = 8
23 Correct 1 ms 204 KB n = 8
24 Correct 1 ms 204 KB n = 8
25 Correct 1 ms 204 KB n = 8
26 Correct 1 ms 204 KB n = 8
27 Correct 1 ms 204 KB n = 8
28 Correct 1 ms 204 KB n = 8
29 Correct 1 ms 204 KB n = 16
30 Correct 1 ms 204 KB n = 16
31 Correct 1 ms 204 KB n = 16
32 Correct 1 ms 204 KB n = 16
33 Correct 1 ms 204 KB n = 16
34 Correct 1 ms 204 KB n = 16
35 Correct 1 ms 204 KB n = 16
36 Correct 1 ms 204 KB n = 15
37 Correct 0 ms 204 KB n = 8
38 Correct 1 ms 204 KB n = 16
39 Correct 1 ms 204 KB n = 16
40 Correct 1 ms 204 KB n = 9
41 Correct 1 ms 204 KB n = 16
42 Correct 1 ms 204 KB n = 16
43 Correct 1 ms 204 KB n = 16
44 Correct 1 ms 204 KB n = 9
45 Correct 1 ms 204 KB n = 15
46 Correct 1 ms 204 KB n = 16
47 Correct 0 ms 204 KB n = 16
48 Correct 1 ms 204 KB n = 16
49 Correct 1712 ms 45776 KB n = 199999
50 Correct 1888 ms 49568 KB n = 199991
51 Correct 1567 ms 49544 KB n = 199993
52 Correct 1118 ms 38072 KB n = 152076
53 Correct 622 ms 23824 KB n = 93249
54 Correct 1090 ms 39804 KB n = 199910
55 Correct 1168 ms 48828 KB n = 199999
56 Correct 937 ms 40008 KB n = 199997
57 Correct 1449 ms 41416 KB n = 171294
58 Correct 1014 ms 36140 KB n = 140872
59 Correct 993 ms 40188 KB n = 199886
60 Correct 1326 ms 49560 KB n = 199996
61 Correct 925 ms 41860 KB n = 200000
62 Correct 1292 ms 53092 KB n = 199998
63 Correct 1022 ms 50344 KB n = 200000
64 Correct 1528 ms 51448 KB n = 199998
65 Correct 1343 ms 59672 KB n = 200000
66 Correct 1515 ms 49032 KB n = 190000
67 Correct 1235 ms 54476 KB n = 177777
68 Correct 511 ms 24600 KB n = 100000
69 Correct 1604 ms 48992 KB n = 200000
70 Correct 1205 ms 49992 KB n = 200000
71 Correct 1408 ms 49072 KB n = 200000
72 Correct 1704 ms 49532 KB n = 200000
73 Correct 1796 ms 49776 KB n = 200000
74 Correct 1719 ms 49828 KB n = 200000
75 Correct 1536 ms 50272 KB n = 200000
76 Correct 1516 ms 50288 KB n = 200000
77 Correct 481 ms 37324 KB n = 200000
78 Correct 353 ms 28556 KB n = 200000
79 Correct 1696 ms 46768 KB n = 184307
80 Correct 562 ms 19328 KB n = 76040
81 Correct 986 ms 40288 KB n = 199981
82 Correct 1248 ms 49332 KB n = 199994
83 Correct 967 ms 41844 KB n = 199996
84 Correct 1258 ms 52812 KB n = 199998
85 Correct 1123 ms 50380 KB n = 200000
86 Correct 1461 ms 51604 KB n = 199998
87 Correct 1753 ms 59624 KB n = 200000
88 Correct 1568 ms 52216 KB n = 200000
89 Correct 1334 ms 59676 KB n = 200000
90 Correct 1436 ms 48824 KB n = 200000
91 Correct 1371 ms 50036 KB n = 200000
92 Correct 1650 ms 49036 KB n = 200000