Submission #1011037

#	Time	Username	Problem	Language	Result	Execution time	Memory
1011037		Dan4Life	Longest Trip (IOI23_longesttrip)	C++17	70 / 100	24 ms	684 KiB

longesttrip

#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){
	srand(time(0));
	
	deque<int> line1, line2; line1.clear(),line2.clear();
	vector<int> ord(N,0); iota(all(ord),0);
	random_shuffle(all(ord));
	for(auto x : ord){
		if(line1.empty()) line1.push_back(x);
		else if(are_connected({x},{line1.back()})) line1.push_back(x);
		else if(!line2.empty() and are_connected({line1.back()},{line2.back()})){
			reverse(all(line2));
			for(auto u : line2) line1.pb(u);
			line2.clear(); line2.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;
	}
	if(are_connected({line1.back()},{line2[0]})){
		for(auto u : line1) ans.pb(u);
		for(auto u : line2) ans.pb(u);
		return ans;
	}
	if(are_connected({line1[0]},{line2[0]})){
		reverse(all(line2));
		for(auto u : line2) ans.pb(u);
		for(auto u : line1) ans.pb(u);
		return ans;
	}
	if(are_connected({line1[0]},{line2.back()})){
		for(auto u : line2) ans.pb(u);
		for(auto u : line1) 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;
	deque<int> cur; cur.clear();
	for(int i = 0; i < sz(line1); i++){
		cur.pb(line1[i]);
		if(connected(cur,line2)){
			x=i; break;
		}
	}
	/*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;
	vector<int> cur2; cur2.clear();
	for(int i = 0; i < sz(line2); i++){
		cur2.pb(line2[i]);
		if(are_connected({line1[x]},cur2)){
			y=i; break;
		}
	}
	for(int i = x+1; i<sz(line1); i++) ans.pb(line1[i]);
	for(int i = 0; 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;
}

Compilation message (stderr)

longesttrip.cpp: In function 'std::vector<int> longest_trip(int, int)':
longesttrip.cpp:64:13: warning: unused variable 'r' [-Wunused-variable]
   64 |  int l = 0, r = sz(line1)-1;
      |             ^

• Subtask 0: 0 / 0
• Subtask 1: 5 / 5
• Subtask 2: 10 / 10
• Subtask 3: 25 / 25
• Subtask 4: 30 / 60