Submission #588235

# Submission time Handle Problem Language Result Execution time Memory
588235 2022-07-02T20:41:19 Z SlavicG Roller Coaster Railroad (IOI16_railroad) C++17
100 / 100
181 ms 11560 KB
#include "bits/stdc++.h"
#include "railroad.h"

using namespace std;

const int N = 4e5 + 10;
int p[N], balance[N];
int get(int a) {
    return p[a] = (p[a] == a ? a : get(p[a]));
}
int uni(int a, int b) {
    a = get(a), b = get(b);
    if(a == b) return 0;
    p[a] = b;
    return 1;
}

long long plan_roller_coaster(std::vector<int> s, std::vector<int> t) {
    for(int i = 0; i < N; ++i) p[i] = i;
    s.push_back((int)1e9);
    t.push_back(1);
    vector<int> c;
    for(int i = 0; i < s.size(); ++i) c.push_back(s[i]), c.push_back(t[i]);
    sort(c.begin(), c.end());
    c.erase(unique(c.begin(), c.end()), c.end());
    for(int i = 0; i < (int)s.size(); ++i) {
        s[i] = lower_bound(c.begin(), c.end(), s[i]) - c.begin();
        t[i] = lower_bound(c.begin(), c.end(), t[i]) - c.begin();

        ++balance[s[i]], --balance[t[i]];
        uni(s[i], t[i]);
    }

    long long ans = 0;
    vector<pair<int, int>> edges;

    for(int i = 0; i + 1 < (int)c.size(); ++i) {
        balance[i + 1] += balance[i];
        ans += 1LL * (c[i + 1] - c[i]) * max(0, balance[i]);

        if(balance[i] == 0) edges.push_back({c[i + 1] - c[i], i});
        else uni(i, i + 1);
    }

    sort(edges.begin(), edges.end());
    for(auto [c, i] : edges) {
        if(uni(i, i + 1)) {
            ans += c;
        }
    }
    return ans;
}

Compilation message

railroad.cpp: In function 'long long int plan_roller_coaster(std::vector<int>, std::vector<int>)':
railroad.cpp:23:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   23 |     for(int i = 0; i < s.size(); ++i) c.push_back(s[i]), c.push_back(t[i]);
      |                    ~~^~~~~~~~~~
# Verdict Execution time Memory Grader output
1 Correct 1 ms 1748 KB n = 2
2 Correct 1 ms 1748 KB n = 2
3 Correct 1 ms 1748 KB n = 2
4 Correct 1 ms 1748 KB n = 2
5 Correct 1 ms 1748 KB n = 2
6 Correct 1 ms 1748 KB n = 2
7 Correct 1 ms 1748 KB n = 3
8 Correct 1 ms 1748 KB n = 3
9 Correct 1 ms 1748 KB n = 3
10 Correct 1 ms 1876 KB n = 8
11 Correct 2 ms 1832 KB n = 8
12 Correct 1 ms 1748 KB n = 8
13 Correct 1 ms 1748 KB n = 8
14 Correct 1 ms 1748 KB n = 8
15 Correct 1 ms 1748 KB n = 8
16 Correct 1 ms 1748 KB n = 8
17 Correct 1 ms 1748 KB n = 8
18 Correct 1 ms 1796 KB n = 8
19 Correct 1 ms 1748 KB n = 3
20 Correct 1 ms 1748 KB n = 7
21 Correct 1 ms 1748 KB n = 8
22 Correct 1 ms 1748 KB n = 8
23 Correct 1 ms 1876 KB n = 8
24 Correct 2 ms 1748 KB n = 8
25 Correct 1 ms 1748 KB n = 8
# Verdict Execution time Memory Grader output
1 Correct 1 ms 1748 KB n = 2
2 Correct 1 ms 1748 KB n = 2
3 Correct 1 ms 1748 KB n = 2
4 Correct 1 ms 1748 KB n = 2
5 Correct 1 ms 1748 KB n = 2
6 Correct 1 ms 1748 KB n = 2
7 Correct 1 ms 1748 KB n = 3
8 Correct 1 ms 1748 KB n = 3
9 Correct 1 ms 1748 KB n = 3
10 Correct 1 ms 1876 KB n = 8
11 Correct 2 ms 1832 KB n = 8
12 Correct 1 ms 1748 KB n = 8
13 Correct 1 ms 1748 KB n = 8
14 Correct 1 ms 1748 KB n = 8
15 Correct 1 ms 1748 KB n = 8
16 Correct 1 ms 1748 KB n = 8
17 Correct 1 ms 1748 KB n = 8
18 Correct 1 ms 1796 KB n = 8
19 Correct 1 ms 1748 KB n = 3
20 Correct 1 ms 1748 KB n = 7
21 Correct 1 ms 1748 KB n = 8
22 Correct 1 ms 1748 KB n = 8
23 Correct 1 ms 1876 KB n = 8
24 Correct 2 ms 1748 KB n = 8
25 Correct 1 ms 1748 KB n = 8
26 Correct 2 ms 1748 KB n = 8
27 Correct 1 ms 1876 KB n = 8
28 Correct 1 ms 1748 KB n = 8
29 Correct 1 ms 1748 KB n = 16
30 Correct 1 ms 1748 KB n = 16
31 Correct 1 ms 1748 KB n = 16
32 Correct 1 ms 1748 KB n = 16
33 Correct 1 ms 1748 KB n = 16
34 Correct 1 ms 1748 KB n = 16
35 Correct 1 ms 1748 KB n = 16
36 Correct 1 ms 1748 KB n = 15
37 Correct 1 ms 1876 KB n = 8
38 Correct 1 ms 1748 KB n = 16
39 Correct 1 ms 1876 KB n = 16
40 Correct 1 ms 1748 KB n = 9
41 Correct 1 ms 1748 KB n = 16
42 Correct 1 ms 1748 KB n = 16
43 Correct 2 ms 1748 KB n = 16
44 Correct 1 ms 1876 KB n = 9
45 Correct 1 ms 1876 KB n = 15
46 Correct 1 ms 1760 KB n = 16
47 Correct 1 ms 1876 KB n = 16
48 Correct 1 ms 1748 KB n = 16
# Verdict Execution time Memory Grader output
1 Correct 154 ms 9128 KB n = 199999
2 Correct 168 ms 9168 KB n = 199991
3 Correct 160 ms 9172 KB n = 199993
4 Correct 115 ms 7604 KB n = 152076
5 Correct 70 ms 5316 KB n = 93249
6 Correct 143 ms 8876 KB n = 199910
7 Correct 148 ms 9380 KB n = 199999
8 Correct 136 ms 9124 KB n = 199997
9 Correct 141 ms 8332 KB n = 171294
10 Correct 104 ms 6316 KB n = 140872
11 Correct 136 ms 8908 KB n = 199886
12 Correct 171 ms 9528 KB n = 199996
13 Correct 143 ms 9272 KB n = 200000
14 Correct 151 ms 11172 KB n = 199998
15 Correct 152 ms 11064 KB n = 200000
16 Correct 172 ms 11192 KB n = 199998
17 Correct 147 ms 11444 KB n = 200000
18 Correct 135 ms 8952 KB n = 190000
19 Correct 127 ms 10536 KB n = 177777
20 Correct 78 ms 5416 KB n = 100000
21 Correct 173 ms 9272 KB n = 200000
22 Correct 174 ms 11188 KB n = 200000
23 Correct 172 ms 11208 KB n = 200000
# Verdict Execution time Memory Grader output
1 Correct 1 ms 1748 KB n = 2
2 Correct 1 ms 1748 KB n = 2
3 Correct 1 ms 1748 KB n = 2
4 Correct 1 ms 1748 KB n = 2
5 Correct 1 ms 1748 KB n = 2
6 Correct 1 ms 1748 KB n = 2
7 Correct 1 ms 1748 KB n = 3
8 Correct 1 ms 1748 KB n = 3
9 Correct 1 ms 1748 KB n = 3
10 Correct 1 ms 1876 KB n = 8
11 Correct 2 ms 1832 KB n = 8
12 Correct 1 ms 1748 KB n = 8
13 Correct 1 ms 1748 KB n = 8
14 Correct 1 ms 1748 KB n = 8
15 Correct 1 ms 1748 KB n = 8
16 Correct 1 ms 1748 KB n = 8
17 Correct 1 ms 1748 KB n = 8
18 Correct 1 ms 1796 KB n = 8
19 Correct 1 ms 1748 KB n = 3
20 Correct 1 ms 1748 KB n = 7
21 Correct 1 ms 1748 KB n = 8
22 Correct 1 ms 1748 KB n = 8
23 Correct 1 ms 1876 KB n = 8
24 Correct 2 ms 1748 KB n = 8
25 Correct 1 ms 1748 KB n = 8
26 Correct 2 ms 1748 KB n = 8
27 Correct 1 ms 1876 KB n = 8
28 Correct 1 ms 1748 KB n = 8
29 Correct 1 ms 1748 KB n = 16
30 Correct 1 ms 1748 KB n = 16
31 Correct 1 ms 1748 KB n = 16
32 Correct 1 ms 1748 KB n = 16
33 Correct 1 ms 1748 KB n = 16
34 Correct 1 ms 1748 KB n = 16
35 Correct 1 ms 1748 KB n = 16
36 Correct 1 ms 1748 KB n = 15
37 Correct 1 ms 1876 KB n = 8
38 Correct 1 ms 1748 KB n = 16
39 Correct 1 ms 1876 KB n = 16
40 Correct 1 ms 1748 KB n = 9
41 Correct 1 ms 1748 KB n = 16
42 Correct 1 ms 1748 KB n = 16
43 Correct 2 ms 1748 KB n = 16
44 Correct 1 ms 1876 KB n = 9
45 Correct 1 ms 1876 KB n = 15
46 Correct 1 ms 1760 KB n = 16
47 Correct 1 ms 1876 KB n = 16
48 Correct 1 ms 1748 KB n = 16
49 Correct 154 ms 9128 KB n = 199999
50 Correct 168 ms 9168 KB n = 199991
51 Correct 160 ms 9172 KB n = 199993
52 Correct 115 ms 7604 KB n = 152076
53 Correct 70 ms 5316 KB n = 93249
54 Correct 143 ms 8876 KB n = 199910
55 Correct 148 ms 9380 KB n = 199999
56 Correct 136 ms 9124 KB n = 199997
57 Correct 141 ms 8332 KB n = 171294
58 Correct 104 ms 6316 KB n = 140872
59 Correct 136 ms 8908 KB n = 199886
60 Correct 171 ms 9528 KB n = 199996
61 Correct 143 ms 9272 KB n = 200000
62 Correct 151 ms 11172 KB n = 199998
63 Correct 152 ms 11064 KB n = 200000
64 Correct 172 ms 11192 KB n = 199998
65 Correct 147 ms 11444 KB n = 200000
66 Correct 135 ms 8952 KB n = 190000
67 Correct 127 ms 10536 KB n = 177777
68 Correct 78 ms 5416 KB n = 100000
69 Correct 173 ms 9272 KB n = 200000
70 Correct 174 ms 11188 KB n = 200000
71 Correct 172 ms 11208 KB n = 200000
72 Correct 157 ms 9172 KB n = 200000
73 Correct 181 ms 9252 KB n = 200000
74 Correct 151 ms 9128 KB n = 200000
75 Correct 142 ms 9140 KB n = 200000
76 Correct 139 ms 9136 KB n = 200000
77 Correct 127 ms 10680 KB n = 200000
78 Correct 175 ms 10532 KB n = 200000
79 Correct 144 ms 8668 KB n = 184307
80 Correct 53 ms 4340 KB n = 76040
81 Correct 135 ms 9000 KB n = 199981
82 Correct 138 ms 9512 KB n = 199994
83 Correct 135 ms 9268 KB n = 199996
84 Correct 161 ms 11064 KB n = 199998
85 Correct 151 ms 11064 KB n = 200000
86 Correct 155 ms 11204 KB n = 199998
87 Correct 150 ms 11464 KB n = 200000
88 Correct 146 ms 9544 KB n = 200000
89 Correct 172 ms 11560 KB n = 200000
90 Correct 144 ms 9136 KB n = 200000
91 Correct 171 ms 11184 KB n = 200000
92 Correct 177 ms 11176 KB n = 200000