#include "longesttrip.h"
#include <bits/stdc++.h>
using namespace std;
#define all(v) v.begin(), v.end()
std::vector<int> longest_trip(int n, int d){
    vector<int> P;
    vector<int> Q;
    for(int i = 0;i < n;i++){
        if(P.empty() || are_connected({P.back()}, {i}))P.push_back(i);
        else if(Q.empty() || are_connected({Q.back()}, {i}))Q.push_back(i);     
        else{
            while(P.size())Q.push_back(P.back()), P.pop_back();
            P.push_back(i);
        }   
    }
    if(P.empty() || Q.empty() || !are_connected(P, Q))return (P.size() > Q.size() ? P : Q);
    
    for(int i = 0;i < 2;i++){
        for(int j = 0;j < 2;j++){
            if(are_connected({P.back()}, {Q.front()})){
                for(auto u: Q)P.push_back(u);
                return P;
            }
            reverse(all(Q));
        }
        reverse(all(P));
    }
    assert(0);
    int lo = -1, hi = P.size();
    while(hi - lo > 1){
        int mid = (lo + hi) / 2;
        vector<int> t;
        for(int i = 0;i <= mid;i++)t.push_back(P[i]);
        if(are_connected(t, Q))hi = mid;
        else lo = mid;
    }
    int mi = hi;
    lo = -1, hi = Q.size();
    while(hi - lo > 1){
        int mid = (lo + hi) / 2;
        vector<int> t;
        for(int i = 0;i <= mid;i++)t.push_back(Q[i]);
        if(are_connected({P[mi]}, t))hi = mid;
        else lo = mid;
    }
    int mj = hi;
    vector<int> ans;
    for(int i = mi + 1;i < P.size();i++)ans.push_back(P[i]);
    for(int i = 0;i <= mi;i++)ans.push_back(P[i]);
    for(int i = mj;mj < Q.size();mj++)ans.push_back(Q[i]);
    for(int i = 0;i < mj;i++)ans.push_back(Q[i]);
    return ans;
}
| # | Verdict | Execution time | Memory | Grader output | 
|---|
| Fetching results... | 
| # | Verdict | Execution time | Memory | Grader output | 
|---|
| Fetching results... | 
| # | Verdict | Execution time | Memory | Grader output | 
|---|
| Fetching results... | 
| # | Verdict | Execution time | Memory | Grader output | 
|---|
| Fetching results... | 
| # | Verdict | Execution time | Memory | Grader output | 
|---|
| Fetching results... |