oj.uz

Submission #556750

# Submission time Handle Problem Language Result Execution time Memory
556750 2022-05-03T21:34:04 Z timreizin Teleporters (IOI08_teleporters) C++17
95 / 100
 1000 ms 36104 KB 
#include <iostream>
#include <vector>
#include <algorithm>
#include <deque>
#include <set>
#include <string>
#include <numeric>
#include <cmath>
#include <cassert>
#include <array>
#include <numeric>

using namespace std;

using ll = long long;

int dfs2(int v, vector<int> &adj, vector<bool> &used)
{
    used[v] = true;
    if (adj[v] == -1 || used[adj[v]]) return 1;
    return dfs2(adj[v], adj, used) + 1;
}

int main()
{
    cin.tie(0)->sync_with_stdio(0);
    int n, m;
    cin >> n >> m;
    vector<pair<int, int>> we(n);
    vector<pair<int, int>> closest(2 * n);
    for (int i = 0; i < n; ++i)
    {
        cin >> we[i].first >> we[i].second;
        closest[i * 2] = {we[i].first, i * 2};
        closest[i * 2 + 1] = {we[i].second, i * 2 + 1};
    }
    sort(closest.begin(), closest.end());
    vector<int> adj(2 * n, -1);
    for (int i = 0; i < n; ++i)
    {
        auto it = upper_bound(closest.begin(), closest.end(), pair(we[i].second, (int)1e9));
        if (it != closest.end()) adj[2 * i] = it->second;
        it = upper_bound(closest.begin(), closest.end(), pair(we[i].first, (int)1e9));
        if (it != closest.end()) adj[2 * i + 1] = it->second;
    }
    int start = closest.front().second;
    vector<bool> used(2 * n);
    auto dfs = [&adj, &used](auto self, int v) -> int
    {
        used[v] = true;
        if (adj[v] == -1 || used[adj[v]]) return 1;
        return self(self, adj[v]) + 1;
    };
    int res = dfs2(start, adj, used);
    vector<int> comps;
    for (int i = 0; i < 2 * n; ++i) if (!used[i]) comps.push_back(dfs2(i, adj, used));
    sort(comps.rbegin(), comps.rend());
    for (int i = 0; i < min(m, (int)comps.size()); ++i) res += comps[i] + 2;
    m -= (int)comps.size();
    m = max(m, 0);
    res += 2 * (m - m % 2);
    res += m % 2;
    cout << res;
    return 0;
}

Compilation message

teleporters.cpp: In function 'int main()':
teleporters.cpp:48:10: warning: variable 'dfs' set but not used [-Wunused-but-set-variable]
   48 |     auto dfs = [&adj, &used](auto self, int v) -> int
      |          ^~~

Subtask #1
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Output is correct

Subtask #2
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Output is correct

Subtask #3
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Output is correct

Subtask #4
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Output is correct

Subtask #5
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Output is correct

Subtask #6
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Output is correct

Subtask #7
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Output is correct

Subtask #8
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Output is correct

Subtask #9
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct

Subtask #10
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct

Subtask #11
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 340 KB Output is correct

Subtask #12
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 340 KB Output is correct
2 Correct 7 ms 704 KB Output is correct

Subtask #13
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 340 KB Output is correct
2 Correct 7 ms 708 KB Output is correct
3 Correct 19 ms 1172 KB Output is correct

Subtask #14
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 11 ms 724 KB Output is correct

Subtask #15
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 9 ms 852 KB Output is correct

Subtask #16
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 86 ms 4672 KB Output is correct
2 Correct 233 ms 10416 KB Output is correct

Subtask #17
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 140 ms 7624 KB Output is correct
2 Correct 342 ms 15040 KB Output is correct

Subtask #18
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 504 ms 22112 KB Output is correct
2 Correct 627 ms 26152 KB Output is correct
3 Correct 689 ms 30404 KB Output is correct

Subtask #19
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 804 ms 30152 KB Output is correct
2 Correct 914 ms 32584 KB Output is correct
3 Correct 870 ms 31456 KB Output is correct
4 Correct 905 ms 31884 KB Output is correct

Subtask #20
0 / 5.0

# Verdict Execution time Memory Grader output
1 Execution timed out 1040 ms 36104 KB Time limit exceeded
2 Halted 0 ms 0 KB -