oj.uz

Submission #899641

# Submission time Handle Problem Language Result Execution time Memory
899641 2024-01-06T16:27:52 Z Szil Longest Trip (IOI23_longesttrip) C++17
15 / 100
 12 ms 856 KB 
#include <bits/stdc++.h>
#include "longesttrip.h"

using namespace std;

vector<int> solve3(int N) {
    vector<int> ans(N);
    iota(ans.begin(), ans.end(), 0);
    return ans;
}

vector<int> solve2(int N) {
    queue<int> q;
    vector<int> ans;
    for (int i = 0; i < N; i++) {
        q.push(i);
    }
    int extra = -1;
    while (!q.empty()) {
        int u = q.front();
        q.pop();
        ans.emplace_back(u);
        if (q.empty()) continue;

        int v = q.front();
        if (!are_connected({u}, {v})) {
            q.pop();
            if (q.empty()) {
                extra = v;
            } else {
                q.push(v);
            }
        }
    }
    vector<int> res;
    for (int i = 0; i < ans.size(); i++) {
        if (i == 0 && extra != -1) res.emplace_back(extra);
        res.emplace_back(ans[i]);
    }
    for (int i = 1; i < res.size(); i++) {
        assert(are_connected({res[i-1]}, {res[i]}));
    }
    assert(res.size() == N);
    return res;
}

vector<int> merge_vectors(const vector<int> &a, const vector<int> &b) {
    vector<int> res;
    for (int i : a) res.emplace_back(i);
    for (int i : b) res.emplace_back(i);
    return res;
}

vector<int> get_interval(const vector<int> &a, int l, int r) {
    vector<int> res;
    for (int i = l; i <= r; i++) {
        res.emplace_back(a[i]);
    }
    return res;
}

int find_connection(const vector<int> &a, const vector<int> &b) {
    int lo = 0, hi = a.size() - 1;
    while (lo < hi) {
        int mid = (lo + hi) / 2;
        if (are_connected(get_interval(a, lo, mid), b)) {
            hi = mid;
        } else {
            lo = mid + 1;
        }
    }
    return lo;
}

vector<int> add_to_component(vector<int> comp, int u) {
    if (comp.empty()) return {u};
    if (comp.size() == 1) return {comp.front(), u};

    // TODO remove this and only check with binary search
    if (are_connected({comp.back()}, {u})) {
        return merge_vectors(comp, {u});
    }

    int idx = find_connection(comp, {u});

    if (idx == 0) {
        return merge_vectors({u}, comp);
    }

    vector<int> res1 = {u, comp[idx]};
    auto res2 = merge_vectors(res1, get_interval(comp, idx+1, comp.size()-1));
    auto res3 = merge_vectors(res2, get_interval(comp, 0, idx-1));

    return res3;
}

vector<int> merge_components(vector<int> &a, vector<int> &b) {
    if (are_connected({a.front()}, {b.front()})) {
        reverse(a.begin(), a.end());
        return merge_vectors(a, b);
    }
    if (are_connected({a.back()}, {b.front()})) {
        return merge_vectors(a, b);
    }
    if (are_connected({a.front()}, {b.back()})) {
        reverse(a.begin(), a.end());
        reverse(b.begin(), b.end());
        return merge_vectors(a, b);
    }
    if (are_connected({a.back()}, {b.back()})) {
        reverse(b.begin(), b.end());
        return merge_vectors(a, b);
    }

    int idx_in_a = find_connection(a, b);
    int idx_in_b = find_connection(b, a);

    auto res1 = get_interval(a, idx_in_a+1, a.size()-1);
    auto res2 = merge_vectors(res1, get_interval(a, 0, idx_in_a));
    auto res3 = merge_vectors(res2, get_interval(b, idx_in_b, b.size()-1));
    auto res4 = merge_vectors(res3, get_interval(b, 0, idx_in_b-1));

    return res4;
}

using S = pair<int, vector<int>>;

bool operator<(const S &a, const S &b) {
    return a.first < b.first;
}

vector<int> solve1(int N) {
    priority_queue<S, vector<S>, greater<S>> pq;
    vector<S> done;
    for (int i = 0; i < N; i++) {
        pq.push({1, {i}});
    }
    while (!pq.empty()) {
        S x = pq.top(); pq.pop();
        stack<S> st;
        vector<int> res;
        while (!pq.empty()) {
            S y = pq.top(); pq.pop();
            if (are_connected(x.second, y.second)) {
                res = merge_components(x.second, y.second);
                break;
            } else {
                st.push(y);
            }
        }
        if (res.empty()) done.emplace_back(x);
        else pq.push({res.size(), res});
        while (!st.empty()) {
            pq.push(st.top());
            st.pop();
        }
    }
    sort(done.begin(), done.end());
    return done.back().second;
}

vector<int> longest_trip(int N, int D)
{
    
    if (D == 3) return solve3(N);
    if (D == 2) return solve2(N);
    if (D == 1) return solve1(N);
}

Compilation message

longesttrip.cpp: In function 'std::vector<int> solve2(int)':
longesttrip.cpp:36:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   36 |     for (int i = 0; i < ans.size(); i++) {
      |                     ~~^~~~~~~~~~~~
longesttrip.cpp:40:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   40 |     for (int i = 1; i < res.size(); i++) {
      |                     ~~^~~~~~~~~~~~
In file included from /usr/include/c++/10/cassert:44,
                 from /usr/include/x86_64-linux-gnu/c++/10/bits/stdc++.h:33,
                 from longesttrip.cpp:1:
longesttrip.cpp:43:23: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   43 |     assert(res.size() == N);
      |            ~~~~~~~~~~~^~~~
longesttrip.cpp: In function 'std::vector<int> longest_trip(int, int)':
longesttrip.cpp:168:1: warning: control reaches end of non-void function [-Wreturn-type]
  168 | }
      | ^

Subtask #0
0.0 / 0.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 344 KB Output is correct
2 Correct 3 ms 600 KB Output is correct

Subtask #1
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 344 KB Output is correct
2 Correct 1 ms 344 KB Output is correct
3 Correct 1 ms 344 KB Output is correct
4 Correct 0 ms 344 KB Output is correct
5 Correct 0 ms 344 KB Output is correct

Subtask #2
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 8 ms 344 KB Output is correct
2 Correct 11 ms 596 KB Output is correct
3 Correct 9 ms 344 KB Output is correct
4 Correct 9 ms 344 KB Output is correct
5 Correct 10 ms 344 KB Output is correct
6 Correct 10 ms 344 KB Output is correct
7 Correct 9 ms 344 KB Output is correct
8 Correct 8 ms 344 KB Output is correct
9 Correct 9 ms 344 KB Output is correct
10 Correct 10 ms 344 KB Output is correct
11 Correct 8 ms 344 KB Output is correct
12 Correct 11 ms 600 KB Output is correct
13 Correct 8 ms 344 KB Output is correct

Subtask #3
0.0 / 25.0

# Verdict Execution time Memory Grader output
1 Correct 9 ms 344 KB Output is correct
2 Correct 10 ms 344 KB Output is correct
3 Correct 12 ms 344 KB Output is correct
4 Correct 10 ms 600 KB Output is correct
5 Correct 9 ms 604 KB Output is correct
6 Correct 10 ms 344 KB Output is correct
7 Correct 9 ms 344 KB Output is correct
8 Correct 11 ms 344 KB Output is correct
9 Correct 10 ms 344 KB Output is correct
10 Correct 11 ms 604 KB Output is correct
11 Correct 11 ms 604 KB Output is correct
12 Correct 9 ms 604 KB Output is correct
13 Correct 10 ms 604 KB Output is correct
14 Correct 8 ms 344 KB Output is correct
15 Correct 11 ms 344 KB Output is correct
16 Incorrect 1 ms 344 KB Incorrect
17 Halted 0 ms 0 KB -

Subtask #4
0.0 / 60.0

# Verdict Execution time Memory Grader output
1 Correct 8 ms 344 KB Output is correct
2 Correct 8 ms 344 KB Output is correct
3 Correct 9 ms 344 KB Output is correct
4 Correct 10 ms 344 KB Output is correct
5 Partially correct 10 ms 604 KB Output is partially correct
6 Correct 10 ms 344 KB Output is correct
7 Correct 10 ms 344 KB Output is correct
8 Correct 12 ms 344 KB Output is correct
9 Correct 11 ms 600 KB Output is correct
10 Partially correct 10 ms 856 KB Output is partially correct
11 Partially correct 11 ms 600 KB Output is partially correct
12 Partially correct 10 ms 604 KB Output is partially correct
13 Partially correct 9 ms 456 KB Output is partially correct
14 Correct 10 ms 344 KB Output is correct
15 Correct 10 ms 344 KB Output is correct
16 Incorrect 1 ms 344 KB Incorrect
17 Halted 0 ms 0 KB -