#include "longesttrip.h"
#include <bits/stdc++.h>
using namespace std;
bool connected(vector<int>A,vector<int>B){
if (A.empty() || B.empty()) return false;
sort(A.begin(),A.end());
sort(B.begin(),B.end());
A.erase(unique(A.begin(),A.end()),A.end());
B.erase(unique(B.begin(),B.end()),B.end());
return are_connected(A,B);
}
std::vector<int> longest_trip(int N, int D)
{
vector<int>A={0},B;
bool not_connected = false;
for (int i=1;i<N;i++){
if (connected({A.back()},{i})){
A.push_back(i);
} else {
if (B.empty() || not_connected || connected({B.back()},{i})){
B.push_back(i);
not_connected=true;
} else {
reverse(B.begin(),B.end());
for (auto i:B) A.push_back(i);
B={i};
}
}
}
if (B.empty()) return A;
if (!connected(A,B)){
if (A.size()>B.size()) return A;
else return B;
}
if (connected({A[0],A.back()},{B[0],B.back()})){
if (connected({A[0]},{B[0],B.back()})){
if (connected({A[0]},{B[0]})){
reverse(B.begin(),B.end());
for (auto i:A){
B.push_back(i);
}
return B;
} else {
for (auto i:A){
B.push_back(i);
}
return B;
}
} else {
if (connected({A.back()},{B[0]})){
for (auto i:B){
A.push_back(i);
}
return A;
} else {
reverse(B.begin(),B.end());
for (auto i:B){
A.push_back(i);
}
return A;
}
}
}
vector<int>rem,A0,B0;
{
int pos=0;
for (int i=8;i>=0;i--){
pos+=(1<<i);
if (pos>=A.size()){
pos-=(1<<i);
continue;
}
vector<int>C;
for (int j=0;j<pos;j++) C.push_back(A[j]);
if (connected(C,B)){
pos-=(1<<i);
}
}
rem=A;
A={A[pos]};
swap(A,B);
}
A0=rem;
{
int pos=0;
for (int i=8;i>=0;i--){
pos+=(1<<i);
if (pos>=A.size()){
pos-=(1<<i);
continue;
}
vector<int>C;
for (int j=0;j<pos;j++) C.push_back(A[j]);
if (connected(C,B)){
pos-=(1<<i);
}
}
rem=A;
A={A[pos]};
swap(A,B);
}
B0=rem;
int posA=-1,posB=-1;
for (int i=0;i<A0.size();i++){
if (A0[i]==A[0]) posA=i;
}
for (int i=0;i<B0.size();i++){
if (B0[i]==B[0]) posB=i;
}
vector<int>ans;
for (int i=posA+1;i<A0.size();i++){
ans.push_back(A0[i]);
}
for (int i=0;i<=posA;i++){
ans.push_back(A0[i]);
}
for (int i=posB;i<B0.size();i++){
ans.push_back(B0[i]);
}
for (int i=0;i<posB;i++){
ans.push_back(B0[i]);
}
// set<int>st;
// for (auto i:ans){
// st.insert(i);
// }
// if (st.size()<ans.size()) exit(1);
return ans;
}
Compilation message
longesttrip.cpp: In function 'std::vector<int> longest_trip(int, int)':
longesttrip.cpp:70:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
70 | if (pos>=A.size()){
| ~~~^~~~~~~~~~
longesttrip.cpp:89:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
89 | if (pos>=A.size()){
| ~~~^~~~~~~~~~
longesttrip.cpp:107:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
107 | for (int i=0;i<A0.size();i++){
| ~^~~~~~~~~~
longesttrip.cpp:110:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
110 | for (int i=0;i<B0.size();i++){
| ~^~~~~~~~~~
longesttrip.cpp:114:24: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
114 | for (int i=posA+1;i<A0.size();i++){
| ~^~~~~~~~~~
longesttrip.cpp:120:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
120 | for (int i=posB;i<B0.size();i++){
| ~^~~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
344 KB |
Output is correct |
2 |
Incorrect |
2 ms |
600 KB |
Incorrect |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
5 ms |
344 KB |
Output is correct |
2 |
Correct |
6 ms |
344 KB |
Output is correct |
3 |
Correct |
5 ms |
344 KB |
Output is correct |
4 |
Correct |
4 ms |
344 KB |
Output is correct |
5 |
Correct |
5 ms |
600 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
5 ms |
344 KB |
Output is correct |
2 |
Correct |
5 ms |
344 KB |
Output is correct |
3 |
Correct |
4 ms |
344 KB |
Output is correct |
4 |
Correct |
5 ms |
340 KB |
Output is correct |
5 |
Correct |
5 ms |
344 KB |
Output is correct |
6 |
Correct |
11 ms |
344 KB |
Output is correct |
7 |
Correct |
6 ms |
344 KB |
Output is correct |
8 |
Correct |
5 ms |
344 KB |
Output is correct |
9 |
Correct |
4 ms |
344 KB |
Output is correct |
10 |
Correct |
5 ms |
440 KB |
Output is correct |
11 |
Correct |
5 ms |
344 KB |
Output is correct |
12 |
Correct |
5 ms |
444 KB |
Output is correct |
13 |
Correct |
6 ms |
440 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
6 ms |
344 KB |
Output is correct |
2 |
Correct |
6 ms |
344 KB |
Output is correct |
3 |
Correct |
5 ms |
344 KB |
Output is correct |
4 |
Correct |
6 ms |
344 KB |
Output is correct |
5 |
Correct |
5 ms |
344 KB |
Output is correct |
6 |
Correct |
13 ms |
344 KB |
Output is correct |
7 |
Correct |
6 ms |
344 KB |
Output is correct |
8 |
Correct |
5 ms |
344 KB |
Output is correct |
9 |
Correct |
6 ms |
344 KB |
Output is correct |
10 |
Correct |
5 ms |
440 KB |
Output is correct |
11 |
Correct |
5 ms |
600 KB |
Output is correct |
12 |
Correct |
6 ms |
600 KB |
Output is correct |
13 |
Correct |
5 ms |
436 KB |
Output is correct |
14 |
Correct |
7 ms |
344 KB |
Output is correct |
15 |
Incorrect |
2 ms |
344 KB |
Incorrect |
16 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
5 ms |
344 KB |
Output is correct |
2 |
Correct |
5 ms |
344 KB |
Output is correct |
3 |
Correct |
6 ms |
344 KB |
Output is correct |
4 |
Correct |
5 ms |
344 KB |
Output is correct |
5 |
Correct |
6 ms |
600 KB |
Output is correct |
6 |
Correct |
10 ms |
344 KB |
Output is correct |
7 |
Correct |
7 ms |
344 KB |
Output is correct |
8 |
Correct |
7 ms |
344 KB |
Output is correct |
9 |
Correct |
6 ms |
600 KB |
Output is correct |
10 |
Correct |
4 ms |
440 KB |
Output is correct |
11 |
Correct |
4 ms |
344 KB |
Output is correct |
12 |
Correct |
8 ms |
600 KB |
Output is correct |
13 |
Correct |
5 ms |
440 KB |
Output is correct |
14 |
Correct |
7 ms |
344 KB |
Output is correct |
15 |
Incorrect |
3 ms |
340 KB |
Incorrect |
16 |
Halted |
0 ms |
0 KB |
- |