#include <bits/stdc++.h>
#include "longesttrip.h"
using namespace std;
struct XorList {
struct Node {
int val;
Node *x, *y;
Node(int val) : val(val), x(NULL), y(NULL) { }
} *start, *end;
void merge(XorList *o) {
if (end->x == NULL) {
end->x = o->start;
} else {
end->y = o->start;
}
if (o->start->x == NULL) {
o->start->x = end;
} else {
o->start->y = end;
}
end = o->end;
o->start = o->end = NULL;
}
vector<int> vectorize() {
vector<int> ans;
Node* curr = start, *old = NULL;
while (curr != NULL) {
ans.push_back(curr->val);
if (old == curr->x) {
old = curr;
curr = old->y;
} else {
old = curr;
curr = old->x;
}
}
return ans;
}
int first() { return start->val; }
int last() { return end->val; }
void reverse() { swap(start, end); }
XorList(int val) { start = end = new Node(val); }
};
vector<int> longest_trip(int N, int D) {
XorList *a = new XorList(0);
XorList *b = new XorList(1);
for (int i = 2; i < N; i++) {
int last_a = a->last();
int last_b = b->last();
XorList *m = new XorList(i);
if (are_connected({last_a}, {last_b})) {
b->reverse();
a->merge(b);
b = m;
} else if (are_connected({last_a}, {i})) {
a->merge(m);
} else {
b->merge(m);
}
}
if (are_connected({a->first()}, {b->first()})) {
a->reverse();
a->merge(b);
return a->vectorize();
}
if (are_connected({a->first()}, {b->last()})) {
a->reverse();
b->reverse();
a->merge(b);
return a->vectorize();
}
if (are_connected({a->last()}, {b->first()})) {
a->merge(b);
return a->vectorize();
}
if (are_connected({a->last()}, {b->last()})) {
b->reverse();
a->merge(b);
return a->vectorize();
}
exit(1);
vector<int> chain_a = a->vectorize();
vector<int> chain_b = b->vectorize();
int l1 = -1, r1 = chain_a.size();
while (l1 + 1 < r1) {
int m = (l1 + r1) / 2;
if (are_connected(vector<int>(chain_a.begin(), chain_a.begin() + m + 1), chain_b)) {
r1 = m;
} else {
l1 = m;
}
}
if (r1 == chain_a.size()) {
return chain_a.size() > chain_b.size() ? chain_a : chain_b;
}
int l2 = -1, r2 = chain_b.size();
while (l2 + 1 < r2) {
int m = (l2 + r2) / 2;
if (are_connected(vector<int>(chain_b.begin(), chain_b.begin() + m + 1), {r1})) {
r2 = m;
} else {
l2 = m;
}
}
rotate(chain_a.begin(), chain_a.begin() + r1 + 1, chain_a.end());
rotate(chain_b.begin(), chain_b.begin() + r2, chain_b.end());
copy(chain_b.begin(), chain_b.end(), back_inserter(chain_a));
return chain_a;
}
Compilation message
longesttrip.cpp: In function 'std::vector<int> longest_trip(int, int)':
longesttrip.cpp:113:12: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
113 | if (r1 == chain_a.size()) {
| ~~~^~~~~~~~~~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
0 ms |
344 KB |
Execution failed because the return code was nonzero |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
6 ms |
1216 KB |
Output is correct |
2 |
Correct |
5 ms |
856 KB |
Output is correct |
3 |
Correct |
5 ms |
1116 KB |
Output is correct |
4 |
Correct |
4 ms |
600 KB |
Output is correct |
5 |
Correct |
5 ms |
864 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
6 ms |
860 KB |
Output is correct |
2 |
Correct |
5 ms |
912 KB |
Output is correct |
3 |
Correct |
4 ms |
892 KB |
Output is correct |
4 |
Correct |
6 ms |
1452 KB |
Output is correct |
5 |
Correct |
4 ms |
728 KB |
Output is correct |
6 |
Correct |
9 ms |
960 KB |
Output is correct |
7 |
Correct |
5 ms |
1120 KB |
Output is correct |
8 |
Correct |
4 ms |
988 KB |
Output is correct |
9 |
Correct |
4 ms |
608 KB |
Output is correct |
10 |
Correct |
4 ms |
628 KB |
Output is correct |
11 |
Correct |
5 ms |
964 KB |
Output is correct |
12 |
Correct |
5 ms |
868 KB |
Output is correct |
13 |
Correct |
4 ms |
880 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
5 ms |
600 KB |
Output is correct |
2 |
Correct |
5 ms |
604 KB |
Output is correct |
3 |
Correct |
6 ms |
1120 KB |
Output is correct |
4 |
Correct |
5 ms |
1112 KB |
Output is correct |
5 |
Correct |
6 ms |
1004 KB |
Output is correct |
6 |
Correct |
7 ms |
868 KB |
Output is correct |
7 |
Correct |
6 ms |
868 KB |
Output is correct |
8 |
Correct |
5 ms |
1124 KB |
Output is correct |
9 |
Correct |
5 ms |
1124 KB |
Output is correct |
10 |
Correct |
5 ms |
976 KB |
Output is correct |
11 |
Correct |
6 ms |
884 KB |
Output is correct |
12 |
Correct |
5 ms |
872 KB |
Output is correct |
13 |
Correct |
5 ms |
644 KB |
Output is correct |
14 |
Runtime error |
0 ms |
344 KB |
Execution failed because the return code was nonzero |
15 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
5 ms |
884 KB |
Output is correct |
2 |
Correct |
5 ms |
600 KB |
Output is correct |
3 |
Correct |
6 ms |
600 KB |
Output is correct |
4 |
Correct |
4 ms |
612 KB |
Output is correct |
5 |
Correct |
5 ms |
1492 KB |
Output is correct |
6 |
Correct |
9 ms |
1140 KB |
Output is correct |
7 |
Correct |
7 ms |
892 KB |
Output is correct |
8 |
Correct |
5 ms |
888 KB |
Output is correct |
9 |
Correct |
5 ms |
612 KB |
Output is correct |
10 |
Correct |
4 ms |
752 KB |
Output is correct |
11 |
Correct |
4 ms |
980 KB |
Output is correct |
12 |
Correct |
5 ms |
884 KB |
Output is correct |
13 |
Correct |
5 ms |
756 KB |
Output is correct |
14 |
Runtime error |
0 ms |
368 KB |
Execution failed because the return code was nonzero |
15 |
Halted |
0 ms |
0 KB |
- |