oj.uz

Submission #1079656

# Submission time Handle Problem Language Result Execution time Memory
1079656 2024-08-28T20:11:00 Z c2zi6 Longest Trip (IOI23_longesttrip) C++17
100 / 100
 19 ms 700 KB 
#define _USE_MATH_DEFINES
#include <bits/stdc++.h>
#define ff first
#define ss second
#define pb push_back
#define all(a) (a).begin(), (a).end()
#define replr(i, a, b) for (int i = int(a); i <= int(b); ++i)
#define reprl(i, a, b) for (int i = int(a); i >= int(b); --i)
#define rep(i, n) for (int i = 0; i < int(n); ++i)
#define mkp(a, b) make_pair(a, b)
using namespace std;
typedef long long ll;
typedef long double ld;
typedef pair<int, int> PII;
typedef vector<int> VI;
typedef vector<PII> VPI;
typedef vector<VI> VVI;
typedef vector<VVI> VVVI;
typedef vector<VPI> VVPI;
typedef pair<ll, ll> PLL;
typedef vector<ll> VL;
typedef vector<PLL> VPL;
typedef vector<VL> VVL;
typedef vector<VVL> VVVL;
typedef vector<VPL> VVPL;
template<class T> T setmax(T& a, T b) {if (a < b) return a = b; return a;}
template<class T> T setmin(T& a, T b) {if (a < b) return a; return a = b;}
#include <ext/pb_ds/assoc_container.hpp>
using namespace __gnu_pbds;
template<class T>
using indset = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
#include "longesttrip.h"

int n;

VI rev(VI a) {
    reverse(all(a));
    return a;
}

// int are_connected(VI a, VI b);

VI longest_trip(int N, int D) {
    n = N;
    VI path1;
    VI path2;

    VI order;
    rep(i, n) order.pb(i);
    mt19937 rng(chrono::steady_clock().now().time_since_epoch().count());
    shuffle(all(order), rng);

    path1.pb(order[0]);
    path2.pb(order[1]);
    bool ardyoq = false;
    // ardyoq es gitem or P-n u Q-n kpac chen?
    replr(i, 2, n-1) {
        int R = order[i];
        int P = path1.back();
        int Q = path2.back();
        if (ardyoq) {
            if (are_connected({P}, {R})) {
                path1.pb(R);
                ardyoq = false;
            } else path2.pb(R);
        } else {
            if (are_connected({P}, {R})) path1.pb(R);
            else if (are_connected({Q}, {R})) {
                path2.pb(R);
                ardyoq = true;
            } else {
                for (int x : rev(path2)) path1.pb(x);
                path2 = VI{R};
            }
        }
    }

    if (are_connected({path2[0]}, {path1[0]})) {
        VI ans;
        for (int x : rev(path2)) ans.pb(x);
        for (int x : path1) ans.pb(x);
        return ans;
    }
    if (are_connected({path2.back()}, {path1[0]})) {
        VI ans;
        for (int x : path2) ans.pb(x);
        for (int x : path1) ans.pb(x);
        return ans;
    }
    if (are_connected({path1.back()}, {path2[0]})) {
        VI ans;
        for (int x : path1) ans.pb(x);
        for (int x : path2) ans.pb(x);
        return ans;
    }

    auto getind2 = [&](int ind1){
        auto ardyoq = [&](int x) {
            VI tmp;
            rep(i, x+1) tmp.pb(path2[i]);
            return are_connected({path1[ind1]}, tmp);
        };
        int l = -1, r = path2.size()-1;
        while (l + 1 < r) {
            int m = (l + r) / 2;
            if (ardyoq(m)) r = m;
            else l = m;
        }
        return r;
    };
    auto getind1 = [&]() {
        auto ardyoq = [&](int x) {
            VI a, b;
            rep(i, x+1) a.pb(path1[i]);
            rep(i, path2.size()) b.pb(path2[i]);
            return are_connected(a, b);
        };
        int l = -1, r = path1.size();
        while (l + 1 < r) {
            int m = (l + r) / 2;
            if (ardyoq(m)) r = m;
            else l = m;
        }
        if (r == path1.size()) return -1;
        return r;
    };

    int ind1 = getind1();
    if (ind1 == -1) {
        if (path1.size() > path2.size()) return path1;
        return path2;
    }
    int ind2 = getind2(ind1);

    VI ans;
    replr(i, ind2+1, ind2+1 + path2.size()-1) {
        ans.pb(path2[i%(int)path2.size()]);
    }
    replr(i, ind1, ind1 + path1.size()-1) {
        ans.pb(path1[i%(int)path1.size()]);
    }
    return ans;
}

Compilation message

longesttrip.cpp: In lambda function:
longesttrip.cpp:124:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  124 |         if (r == path1.size()) return -1;
      |             ~~^~~~~~~~~~~~~~~

Subtask #0
0.0 / 0.0

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

Subtask #1
5.0 / 5.0

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

Subtask #2
10.0 / 10.0

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

Subtask #3
25.0 / 25.0

# Verdict Execution time Memory Grader output
1 Correct 8 ms 344 KB Output is correct
2 Correct 6 ms 344 KB Output is correct
3 Correct 5 ms 344 KB Output is correct
4 Correct 5 ms 344 KB Output is correct
5 Correct 4 ms 344 KB Output is correct
6 Correct 6 ms 344 KB Output is correct
7 Correct 6 ms 344 KB Output is correct
8 Correct 5 ms 344 KB Output is correct
9 Correct 5 ms 344 KB Output is correct
10 Correct 6 ms 344 KB Output is correct
11 Correct 6 ms 344 KB Output is correct
12 Correct 5 ms 344 KB Output is correct
13 Correct 6 ms 344 KB Output is correct
14 Correct 13 ms 344 KB Output is correct
15 Correct 11 ms 344 KB Output is correct
16 Correct 6 ms 344 KB Output is correct
17 Correct 10 ms 344 KB Output is correct
18 Correct 5 ms 344 KB Output is correct
19 Correct 8 ms 344 KB Output is correct
20 Correct 8 ms 344 KB Output is correct
21 Correct 6 ms 340 KB Output is correct
22 Correct 6 ms 344 KB Output is correct
23 Correct 8 ms 344 KB Output is correct
24 Correct 6 ms 344 KB Output is correct
25 Correct 12 ms 344 KB Output is correct
26 Correct 12 ms 344 KB Output is correct
27 Correct 9 ms 344 KB Output is correct
28 Correct 9 ms 344 KB Output is correct
29 Correct 9 ms 344 KB Output is correct
30 Correct 6 ms 344 KB Output is correct
31 Correct 7 ms 344 KB Output is correct
32 Correct 7 ms 344 KB Output is correct
33 Correct 6 ms 344 KB Output is correct
34 Correct 7 ms 344 KB Output is correct
35 Correct 8 ms 344 KB Output is correct
36 Correct 6 ms 344 KB Output is correct
37 Correct 4 ms 344 KB Output is correct
38 Correct 9 ms 600 KB Output is correct
39 Correct 6 ms 600 KB Output is correct
40 Correct 7 ms 444 KB Output is correct
41 Correct 6 ms 344 KB Output is correct
42 Correct 9 ms 440 KB Output is correct
43 Correct 8 ms 600 KB Output is correct
44 Correct 6 ms 344 KB Output is correct
45 Correct 8 ms 344 KB Output is correct
46 Correct 8 ms 344 KB Output is correct
47 Correct 9 ms 344 KB Output is correct
48 Correct 8 ms 344 KB Output is correct
49 Correct 6 ms 344 KB Output is correct
50 Correct 5 ms 344 KB Output is correct
51 Correct 7 ms 344 KB Output is correct
52 Correct 10 ms 344 KB Output is correct
53 Correct 8 ms 440 KB Output is correct
54 Correct 6 ms 344 KB Output is correct
55 Correct 6 ms 344 KB Output is correct
56 Correct 6 ms 344 KB Output is correct
57 Correct 6 ms 592 KB Output is correct
58 Correct 5 ms 344 KB Output is correct
59 Correct 5 ms 344 KB Output is correct
60 Correct 7 ms 600 KB Output is correct
61 Correct 7 ms 344 KB Output is correct
62 Correct 7 ms 344 KB Output is correct
63 Correct 5 ms 344 KB Output is correct
64 Correct 7 ms 444 KB Output is correct

Subtask #4
60.0 / 60.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 344 KB Output is correct
2 Correct 6 ms 344 KB Output is correct
3 Correct 5 ms 344 KB Output is correct
4 Correct 4 ms 344 KB Output is correct
5 Correct 6 ms 344 KB Output is correct
6 Correct 8 ms 344 KB Output is correct
7 Correct 6 ms 344 KB Output is correct
8 Correct 4 ms 344 KB Output is correct
9 Correct 7 ms 344 KB Output is correct
10 Correct 4 ms 344 KB Output is correct
11 Correct 5 ms 344 KB Output is correct
12 Correct 5 ms 344 KB Output is correct
13 Correct 6 ms 344 KB Output is correct
14 Correct 19 ms 344 KB Output is correct
15 Correct 14 ms 344 KB Output is correct
16 Correct 7 ms 344 KB Output is correct
17 Correct 5 ms 344 KB Output is correct
18 Correct 5 ms 344 KB Output is correct
19 Correct 6 ms 344 KB Output is correct
20 Correct 6 ms 344 KB Output is correct
21 Correct 12 ms 344 KB Output is correct
22 Correct 12 ms 344 KB Output is correct
23 Correct 9 ms 344 KB Output is correct
24 Correct 7 ms 344 KB Output is correct
25 Correct 9 ms 344 KB Output is correct
26 Correct 6 ms 344 KB Output is correct
27 Correct 8 ms 344 KB Output is correct
28 Correct 7 ms 344 KB Output is correct
29 Correct 8 ms 344 KB Output is correct
30 Correct 8 ms 344 KB Output is correct
31 Correct 7 ms 344 KB Output is correct
32 Correct 8 ms 344 KB Output is correct
33 Correct 10 ms 344 KB Output is correct
34 Correct 7 ms 344 KB Output is correct
35 Correct 10 ms 344 KB Output is correct
36 Correct 6 ms 344 KB Output is correct
37 Correct 8 ms 344 KB Output is correct
38 Correct 8 ms 344 KB Output is correct
39 Correct 7 ms 344 KB Output is correct
40 Correct 6 ms 440 KB Output is correct
41 Correct 6 ms 344 KB Output is correct
42 Correct 6 ms 344 KB Output is correct
43 Correct 6 ms 344 KB Output is correct
44 Correct 8 ms 344 KB Output is correct
45 Correct 8 ms 344 KB Output is correct
46 Correct 8 ms 340 KB Output is correct
47 Correct 6 ms 344 KB Output is correct
48 Correct 6 ms 444 KB Output is correct
49 Correct 6 ms 440 KB Output is correct
50 Correct 7 ms 448 KB Output is correct
51 Correct 8 ms 604 KB Output is correct
52 Correct 5 ms 344 KB Output is correct
53 Correct 9 ms 600 KB Output is correct
54 Correct 6 ms 344 KB Output is correct
55 Correct 6 ms 344 KB Output is correct
56 Correct 6 ms 444 KB Output is correct
57 Correct 7 ms 344 KB Output is correct
58 Correct 7 ms 344 KB Output is correct
59 Correct 7 ms 344 KB Output is correct
60 Correct 7 ms 344 KB Output is correct
61 Correct 9 ms 600 KB Output is correct
62 Correct 8 ms 452 KB Output is correct
63 Correct 6 ms 452 KB Output is correct
64 Correct 8 ms 344 KB Output is correct
65 Correct 6 ms 344 KB Output is correct
66 Correct 7 ms 444 KB Output is correct
67 Correct 7 ms 600 KB Output is correct
68 Correct 7 ms 344 KB Output is correct
69 Correct 7 ms 444 KB Output is correct
70 Correct 7 ms 444 KB Output is correct
71 Correct 5 ms 700 KB Output is correct
72 Correct 7 ms 344 KB Output is correct
73 Correct 6 ms 452 KB Output is correct
74 Correct 6 ms 344 KB Output is correct
75 Correct 8 ms 344 KB Output is correct
76 Correct 7 ms 344 KB Output is correct
77 Correct 6 ms 452 KB Output is correct
78 Correct 6 ms 344 KB Output is correct
79 Correct 9 ms 448 KB Output is correct
80 Correct 7 ms 448 KB Output is correct
81 Correct 6 ms 344 KB Output is correct
82 Correct 10 ms 344 KB Output is correct