#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);
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
longesttrip.cpp: In function 'std::vector<int> longest_trip(int, int)':
longesttrip.cpp:100:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
100 | for (int i = x + 1; i < a.size(); i++) u.push_back(a[i]);
| ~~^~~~~~~~~~
longesttrip.cpp:102:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
102 | for (int i = y; i < b.size(); i++) u.push_back(b[i]);
| ~~^~~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
344 KB |
Output is correct |
2 |
Correct |
2 ms |
440 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
5 ms |
344 KB |
Output is correct |
2 |
Correct |
11 ms |
344 KB |
Output is correct |
3 |
Correct |
7 ms |
344 KB |
Output is correct |
4 |
Correct |
12 ms |
344 KB |
Output is correct |
5 |
Correct |
11 ms |
344 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
6 ms |
344 KB |
Output is correct |
2 |
Correct |
9 ms |
344 KB |
Output is correct |
3 |
Correct |
7 ms |
344 KB |
Output is correct |
4 |
Correct |
8 ms |
344 KB |
Output is correct |
5 |
Correct |
10 ms |
344 KB |
Output is correct |
6 |
Correct |
9 ms |
344 KB |
Output is correct |
7 |
Correct |
9 ms |
600 KB |
Output is correct |
8 |
Correct |
9 ms |
344 KB |
Output is correct |
9 |
Correct |
12 ms |
340 KB |
Output is correct |
10 |
Correct |
7 ms |
344 KB |
Output is correct |
11 |
Correct |
11 ms |
600 KB |
Output is correct |
12 |
Correct |
8 ms |
344 KB |
Output is correct |
13 |
Correct |
11 ms |
344 KB |
Output is correct |
# |
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 |
12 ms |
344 KB |
Output is correct |
5 |
Correct |
7 ms |
344 KB |
Output is correct |
6 |
Correct |
8 ms |
344 KB |
Output is correct |
7 |
Correct |
9 ms |
344 KB |
Output is correct |
8 |
Correct |
10 ms |
344 KB |
Output is correct |
9 |
Correct |
8 ms |
344 KB |
Output is correct |
10 |
Correct |
8 ms |
600 KB |
Output is correct |
11 |
Correct |
9 ms |
344 KB |
Output is correct |
12 |
Correct |
8 ms |
344 KB |
Output is correct |
13 |
Correct |
8 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 |
Incorrect |
2 ms |
344 KB |
Incorrect |
17 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
6 ms |
344 KB |
Output is correct |
2 |
Correct |
9 ms |
344 KB |
Output is correct |
3 |
Correct |
10 ms |
356 KB |
Output is correct |
4 |
Correct |
10 ms |
344 KB |
Output is correct |
5 |
Correct |
8 ms |
344 KB |
Output is correct |
6 |
Correct |
13 ms |
344 KB |
Output is correct |
7 |
Correct |
11 ms |
344 KB |
Output is correct |
8 |
Correct |
11 ms |
344 KB |
Output is correct |
9 |
Correct |
12 ms |
344 KB |
Output is correct |
10 |
Correct |
11 ms |
344 KB |
Output is correct |
11 |
Correct |
6 ms |
344 KB |
Output is correct |
12 |
Correct |
11 ms |
344 KB |
Output is correct |
13 |
Correct |
5 ms |
344 KB |
Output is correct |
14 |
Correct |
12 ms |
344 KB |
Output is correct |
15 |
Correct |
11 ms |
344 KB |
Output is correct |
16 |
Incorrect |
2 ms |
344 KB |
Incorrect |
17 |
Halted |
0 ms |
0 KB |
- |