Submission #1070914

#	Time	Username	Problem	Language	Result	Execution time	Memory
1070914		zsombor	Longest Trip (IOI23_longesttrip)	C++17	25 / 100	11 ms	600 KiB

longesttrip

#include "longesttrip.h"
#include <bits/stdc++.h>
using namespace std;

int n;
vector<int> a;
vector<int> b;

void solve() {
    for (int i = 2; i < n - 1; i += 2) {
        int x = i, y = i + 1;
        bool A = are_connected({x}, {a.back()});
        bool B = are_connected({x}, {b.back()});
        if (are_connected({x}, {y})) {
            if (A) {
                a.push_back(x);
                a.push_back(y);
            }
            if (!A && B) {
                b.push_back(x);
                b.push_back(y);
            }
            if (!A && !B) {
                while (b.size()) {
                    a.push_back(b.back());
                    b.pop_back();
                }
                b.push_back(x);
                b.push_back(y);
            }
        } else {
            if (A == B) {
                if (!A) swap(x, y);
                a.push_back(x);
                while (b.size()) {
                    a.push_back(b.back());
                    b.pop_back();
                }
                b.push_back(y);
            } else {
                if (!A) swap(x, y);
                a.push_back(x);
                b.push_back(y);
            }
        }
    }
    if (n % 2) {
        if (are_connected({a.back()}, {b.back()})) {
            while (b.size()) {
                a.push_back(b.back());
                b.pop_back();
            }
        }
        if (are_connected({n - 1}, {a.back()})) a.push_back(n - 1);
        else b.push_back(n - 1);
    }
}

bool try_add(vector<int> &v, vector<int> &u) {
    bool A = are_connected({v.back()}, {u[0]});
    bool B = are_connected({v.back()}, {u.back()});
    if (!A && !B) return false;
    if (A) reverse(u.begin(), u.end());
    while (u.size()) {
        v.push_back(u.back());
        u.pop_back();
    }
    return true;
}

bool query(vector<int> &x, int l, int r, vector<int> &v) {
    vector<int> u;
    for (int i = l; i < r; i++) u.push_back(v[i]);
    return are_connected(x, u);
}

int bin(vector<int> x, vector<int> &v) {
    int l = 0, r = v.size(), m;
    while (r - l > 1) {
        m = (l + r) / 2;
        (query(x, l, m, v) ? r = m : l = m);
    }
    return l;
}

vector<int> longest_trip(int N, int D) {
    n = N;
    a = {0};
    b = {1};
    solve();
    if (a.size() < b.size()) swap(a, b);
    return a;
    if (!b.size()) return a;
    if (try_add(a, b)) return a;
    if (try_add(b, a)) return b;
    if (!are_connected(a, b)) return a;

    int x = bin(b, a);
    int y = bin({a[x]}, b);
    vector<int> u;
    for (int i = x + 1; i < a.size(); i++) u.push_back(a[i]);
    for (int i = 0; i <= x; i++) u.push_back(a[i]);
    for (int i = y; i < b.size(); i++) u.push_back(b[i]);
    for (int i = 0; i < y; i++) u.push_back(a[i]);
    return u;
}

Compilation message (stderr)

longesttrip.cpp: In function 'std::vector<int> longest_trip(int, int)':
longesttrip.cpp:101:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  101 |     for (int i = x + 1; i < a.size(); i++) u.push_back(a[i]);
      |                         ~~^~~~~~~~~~
longesttrip.cpp:103:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  103 |     for (int i = y; i < b.size(); i++) u.push_back(b[i]);
      |                     ~~^~~~~~~~~~

• Subtask 0: 0 / 0
• Subtask 1: 0 / 5
• Subtask 2: 0 / 10
• Subtask 3: 25 / 25
• Subtask 4: 0 / 60