#include "longesttrip.h"
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define sz(a) (int)a.size()
#define all(a) begin(a),end(a)
bool connected(deque<int> A, deque<int> B){
vector<int> Av,Bv;Av.clear(),Bv.clear();
for(auto u : A) Av.pb(u);
for(auto u : B) Bv.pb(u);
return are_connected(Av,Bv);
}
vector<int> longest_trip(int N, int D){
vector<int> ord(N,0); iota(all(ord),0);
random_shuffle(all(ord));
deque<int> line1, line2;
line1.push_back(ord[0]);
for(int i = 1; i < N; i++){
int x = ord[i];
if(are_connected({x},{line1.back()})) line1.push_back(x);
else line2.push_back(x);
}
vector<int> ans; ans.clear();
if(line2.empty()){
for(auto u : line1) ans.pb(u);
return ans;
}
if(!connected(line1,line2)){
if(sz(line1)<sz(line2)) swap(line1,line2);
for(auto u : line1) ans.pb(u);
return ans;
}
if(are_connected({line1.back()},{line2.back()})){
for(auto u : line1) ans.pb(u);
reverse(all(line2));
for(auto u : line2) ans.pb(u);
return ans;
}
int l = 0, r = sz(line1)-1;
while(l<r){
int mid = (l+r)/2;
deque<int> dq; dq.clear();
for(int i = 0; i <= mid; i++) dq.pb(line1[i]);
if(connected(dq,line2)) r=mid;
else l=mid+1;
}
int x = l;
l=0, r = sz(line2)-1;
while(l<r){
int mid = (l+r)/2;
vector<int> v; v.clear();
for(int i = 0; i <= mid; i++) v.pb(line2[i]);
if(are_connected({line1[x]},v)) r=mid;
else l=mid+1;
}
int y = l;
for(int i = x-1; i>=0; i--) ans.pb(line1[i]);
for(int i = sz(line1)-1; i>=x; i--) ans.pb(line1[i]);
for(int i = y; i < sz(line2); i++) ans.pb(line2[i]);
for(int i = 0; i < y; i++) ans.pb(line2[i]);
return ans;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
344 KB |
Output is correct |
2 |
Incorrect |
1 ms |
344 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 |
5 ms |
344 KB |
Output is correct |
3 |
Correct |
5 ms |
436 KB |
Output is correct |
4 |
Correct |
4 ms |
344 KB |
Output is correct |
5 |
Correct |
5 ms |
344 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 |
5 ms |
436 KB |
Output is correct |
4 |
Correct |
6 ms |
344 KB |
Output is correct |
5 |
Correct |
5 ms |
344 KB |
Output is correct |
6 |
Correct |
5 ms |
344 KB |
Output is correct |
7 |
Correct |
8 ms |
344 KB |
Output is correct |
8 |
Correct |
7 ms |
344 KB |
Output is correct |
9 |
Correct |
6 ms |
344 KB |
Output is correct |
10 |
Correct |
5 ms |
344 KB |
Output is correct |
11 |
Correct |
6 ms |
344 KB |
Output is correct |
12 |
Correct |
7 ms |
440 KB |
Output is correct |
13 |
Correct |
5 ms |
344 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 |
6 ms |
440 KB |
Output is correct |
4 |
Correct |
5 ms |
344 KB |
Output is correct |
5 |
Correct |
5 ms |
344 KB |
Output is correct |
6 |
Correct |
10 ms |
344 KB |
Output is correct |
7 |
Correct |
6 ms |
344 KB |
Output is correct |
8 |
Correct |
7 ms |
344 KB |
Output is correct |
9 |
Correct |
5 ms |
344 KB |
Output is correct |
10 |
Correct |
5 ms |
344 KB |
Output is correct |
11 |
Correct |
5 ms |
344 KB |
Output is correct |
12 |
Correct |
4 ms |
436 KB |
Output is correct |
13 |
Correct |
8 ms |
344 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 |
6 ms |
344 KB |
Output is correct |
2 |
Correct |
5 ms |
344 KB |
Output is correct |
3 |
Correct |
5 ms |
444 KB |
Output is correct |
4 |
Correct |
6 ms |
344 KB |
Output is correct |
5 |
Correct |
6 ms |
344 KB |
Output is correct |
6 |
Correct |
9 ms |
344 KB |
Output is correct |
7 |
Correct |
6 ms |
344 KB |
Output is correct |
8 |
Correct |
7 ms |
344 KB |
Output is correct |
9 |
Correct |
6 ms |
344 KB |
Output is correct |
10 |
Correct |
5 ms |
344 KB |
Output is correct |
11 |
Correct |
5 ms |
344 KB |
Output is correct |
12 |
Correct |
5 ms |
600 KB |
Output is correct |
13 |
Correct |
4 ms |
344 KB |
Output is correct |
14 |
Correct |
6 ms |
344 KB |
Output is correct |
15 |
Incorrect |
3 ms |
344 KB |
Incorrect |
16 |
Halted |
0 ms |
0 KB |
- |