#include "longesttrip.h"
#include <bits/stdc++.h>
using namespace std;
vector<int> longest_trip(int N, int D) {
vector<int> trip1(1, 0), trip2;
for (int i = 1; i < N; i++) {
bool b = are_connected(vector<int>(1, trip1.front()), vector<int>(1, i)),
b2 = are_connected(vector<int>(1, trip1.back()), vector<int>(1, i));
if (!b && !b2)
trip2.push_back(i);
else if (trip2.empty()) {
if (b)
trip1.insert(trip1.begin(), i);
else
trip2.push_back(i);
} else if (b) {
bool b3 = are_connected(vector<int>(1, trip2.front()), vector<int>(1, i)),
b4 = are_connected(vector<int>(1, trip2.back()), vector<int>(1, i));
if (b3) {
trip1.insert(trip1.begin(), i);
for (auto a : trip2) trip1.insert(trip1.begin(), a);
trip2.clear();
} else if (b4) {
trip2.push_back(i);
for (auto a : trip1) trip2.push_back(a);
trip1 = trip2;
trip2.clear();
} else {
trip1.insert(trip1.begin(), i);
}
} else if (b2) {
bool b3 = are_connected(vector<int>(1, trip2.front()), vector<int>(1, i)),
b4 = are_connected(vector<int>(1, trip2.back()), vector<int>(1, i));
if (b3) {
trip1.push_back(i);
for (auto a : trip2) trip1.push_back(a);
trip2.clear();
} else if (b4) {
reverse(trip2.begin(), trip2.end());
trip1.push_back(i);
for (auto a : trip2) trip1.push_back(a);
trip2.clear();
} else {
trip1.push_back(i);
}
}
}
if (trip2.empty()) return trip1;
if (are_connected(trip1, trip2)) {
if (trip1.size() != 1 && trip2.size() != 1) {
vector<int> vi, vi2;
vi.push_back(trip1.front()), vi.push_back(trip1.back());
vi2.push_back(trip2.front()), vi2.push_back(trip2.back());
if (are_connected(vi, vi2)) {
while (1) {
}
}
}
if (trip1.size() != 1) {
if (!are_connected(vector<int>(1, trip1.front()),
vector<int>(1, trip1.back()))) {
}
while (1) {
}
}
if (trip2.size() != 1) {
if (!are_connected(vector<int>(1, trip2.front()),
vector<int>(1, trip2.back()))) {
}
while (1) {
}
}
int l = 0, r = int(trip1.size()) - 1;
while (l < r) {
int mid = (l + r) / 2;
vector<int> tmp;
for (int i = l; i <= mid; i++) tmp.push_back(trip1[i]);
if (are_connected(tmp, trip2))
r = mid;
else
l = mid + 1;
}
int firstone = l;
l = 0, r = int(trip2.size()) - 1;
while (l < r) {
int mid = (l + r) / 2;
vector<int> tmp;
for (int i = l; i <= mid; i++) tmp.push_back(trip2[i]);
if (are_connected(tmp, vector<int>(1, trip1[firstone])))
r = mid;
else
l = mid + 1;
}
int secondone = l;
vector<int> ans;
for (int i = firstone + 1; i < trip1.size(); i++) ans.push_back(trip1[i]);
for (int i = 0; i <= firstone; i++) ans.push_back(trip1[i]);
for (int i = secondone; i < trip2.size(); i++) ans.push_back(trip2[i]);
for (int i = 0; i < secondone; i++) ans.push_back(trip2[i]);
return ans;
}
if (trip1.size() < trip2.size()) trip1.swap(trip2);
return trip1;
}
Compilation message
longesttrip.cpp: In function 'std::vector<int> longest_trip(int, int)':
longesttrip.cpp:99:34: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
99 | for (int i = firstone + 1; i < trip1.size(); i++) ans.push_back(trip1[i]);
| ~~^~~~~~~~~~~~~~
longesttrip.cpp:101:31: 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 = secondone; i < trip2.size(); i++) ans.push_back(trip2[i]);
| ~~^~~~~~~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
340 KB |
Output is correct |
2 |
Correct |
2 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 |
10 ms |
344 KB |
Output is correct |
4 |
Correct |
7 ms |
344 KB |
Output is correct |
5 |
Correct |
13 ms |
436 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
8 ms |
344 KB |
Output is correct |
2 |
Correct |
14 ms |
344 KB |
Output is correct |
3 |
Correct |
13 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 |
Execution timed out |
3059 ms |
344 KB |
Time limit exceeded |
7 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
5 ms |
344 KB |
Output is correct |
2 |
Correct |
12 ms |
344 KB |
Output is correct |
3 |
Correct |
12 ms |
344 KB |
Output is correct |
4 |
Correct |
9 ms |
344 KB |
Output is correct |
5 |
Correct |
14 ms |
344 KB |
Output is correct |
6 |
Execution timed out |
3059 ms |
344 KB |
Time limit exceeded |
7 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
7 ms |
344 KB |
Output is correct |
2 |
Correct |
9 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 |
Partially correct |
11 ms |
344 KB |
Output is partially correct |
6 |
Execution timed out |
3068 ms |
344 KB |
Time limit exceeded |
7 |
Halted |
0 ms |
0 KB |
- |