#include "longesttrip.h"
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define mp make_pair
#define pii pair<int, int>
#define fi first
#define se second
map<vector<int>, map<vector<int>, int> > mmm;
int query(vector<int> a, vector<int> b){
if (!mmm[a][b]&&!mmm[b][a])mmm[b][a]=mmm[a][b]=are_connected(a, b)+1;
return mmm[a][b]-1;
}
vector<int> longest_trip(int n, int d){
mmm.clear();
if (d==3){
vector<int> ans;
for (int i=0; i<n; ++i)ans.pb(i);
return ans;
}
if (d==2){
deque<int> ans(1, 0);
int done=1;
if (are_connected({0}, {1}))ans.pb(1);
else ans.pb(2), done=2;
for (int i=1; i<n; ++i)if (i!=done){
if (are_connected({i}, {ans.back()}))ans.pb(i);
else ans.push_front(i);
}
vector<int> res;
for (auto a:ans)res.pb(a);
return res;
}
int aa=-1, bb=-1, cc=-1;
for (int i=0; i<n; ++i){
int ooga=-1;
for (int j=0; j<n; ++j){
if (ooga==-1&&i!=j&&query({i}, {j}))ooga=j;
else if (ooga!=-1&&i!=j&&query({i}, {j})){
aa=i, bb=ooga, cc=j;
break;
}
}
}
deque<int> a(1, bb);
for (int i=0; i<n; ++i)if (i!=aa&&i!=bb&&i!=cc){
if (query({i}, {a.back()}))a.pb(i);
else if (query({i}, {b.back()}))b.pb(i);
else{
while (b.size())a.pb(b.back()), b.pop_back();
b.pb(i);
}
}
if (query({i}, {a.back()}))a.pb(i);
else if (query({i}, {b.back()}))b.pb(i);
else{
while (b.size())a.pb(b.back()), b.pop_back();
b.pb(i);
}
if (b.size()>a.size())swap(a, b);
if (query({a.back()}, {b.back()})){
while (b.size())a.pb(b.back()), b.pop_back();
}
else if (query({a.back()}, {b[0]})){
for (auto c:b)a.pb(c);
}
else if (query({a[0]}, {b.back()})){
while (b.size())a.push_front(b.back()), b.pop_back();
}
else if (query({a[0]}, {b[0]})){
for (auto c:b)a.push_front(c);
}
else{
for (int i:b){
if (query({i}, {a.back()}))a.pb(i);
else if (query({i}, {b.back()}))b.pb(i);
else{
for (int j=1; j<n; ++j)if (query({i}, {a[j-1]})&&query({i}, {a[j]})){
a.insert(a.begin()+j, i);
break;
}
}
}
}
vector<int> res;
for (auto c:a)res.pb(c);
return res;
}
Compilation message (stderr)
longesttrip.cpp: In function 'std::vector<int> longest_trip(int, int)':
longesttrip.cpp:52:38: error: 'b' was not declared in this scope
52 | else if (query({i}, {b.back()}))b.pb(i);
| ^
longesttrip.cpp:52:31: error: could not convert '{<expression error>}' from '<brace-enclosed initializer list>' to 'std::vector<int>'
52 | else if (query({i}, {b.back()}))b.pb(i);
| ~~~~~^~~~~~~~~~~~~~~~~
| |
| <brace-enclosed initializer list>
longesttrip.cpp:58:20: error: 'i' was not declared in this scope
58 | if (query({i}, {a.back()}))a.pb(i);
| ^
longesttrip.cpp:58:18: error: could not convert '{<expression error>}' from '<brace-enclosed initializer list>' to 'std::vector<int>'
58 | if (query({i}, {a.back()}))a.pb(i);
| ~~~~~^~~~~~~~~~~~~~~~~
| |
| <brace-enclosed initializer list>
longesttrip.cpp:59:30: error: 'b' was not declared in this scope
59 | else if (query({i}, {b.back()}))b.pb(i);
| ^
longesttrip.cpp:59:23: error: could not convert '{i}' from '<brace-enclosed initializer list>' to 'std::vector<int>'
59 | else if (query({i}, {b.back()}))b.pb(i);
| ~~~~~^~~~~~~~~~~~~~~~~
| |
| <brace-enclosed initializer list>
longesttrip.cpp:64:13: error: 'b' was not declared in this scope
64 | if (b.size()>a.size())swap(a, b);
| ^
longesttrip.cpp:65:32: error: 'b' was not declared in this scope
65 | if (query({a.back()}, {b.back()})){
| ^
longesttrip.cpp:65:18: error: could not convert '{<expression error>}' from '<brace-enclosed initializer list>' to 'std::vector<int>'
65 | if (query({a.back()}, {b.back()})){
| ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
| |
| <brace-enclosed initializer list>
longesttrip.cpp:68:23: error: could not convert '{<expression error>}' from '<brace-enclosed initializer list>' to 'std::vector<int>'
68 | else if (query({a.back()}, {b[0]})){
| ~~~~~^~~~~~~~~~~~~~~~~~~~
| |
| <brace-enclosed initializer list>
longesttrip.cpp:71:23: error: could not convert '{<expression error>}' from '<brace-enclosed initializer list>' to 'std::vector<int>'
71 | else if (query({a[0]}, {b.back()})){
| ~~~~~^~~~~~~~~~~~~~~~~~~~
| |
| <brace-enclosed initializer list>
longesttrip.cpp:74:23: error: could not convert '{<expression error>}' from '<brace-enclosed initializer list>' to 'std::vector<int>'
74 | else if (query({a[0]}, {b[0]})){
| ~~~~~^~~~~~~~~~~~~~~~
| |
| <brace-enclosed initializer list>
longesttrip.cpp:80:39: error: could not convert '{<expression error>}' from '<brace-enclosed initializer list>' to 'std::vector<int>'
80 | else if (query({i}, {b.back()}))b.pb(i);
| ~~~~~^~~~~~~~~~~~~~~~~
| |
| <brace-enclosed initializer list>