Submission #848576

#TimeUsernameProblemLanguageResultExecution timeMemory
848576Edu175Longest Trip (IOI23_longesttrip)C++17
70 / 100
18 ms872 KiB
#include "longesttrip.h"
#include <bits/stdc++.h>
#define pb push_back
#define fst first
#define snd second
#define fore(i,a,b) for(ll i=a,ggdem=b;i<ggdem;++i)
#define SZ(x) ((int)x.size())
#define ALL(x) x.begin(),x.end()
#define mset(a,v) memset((a),(v),sizeof(a))
#define FIN ios::sync_with_stdio(0);cin.tie(0);cout.tie(0)
#define imp(v) for(auto messi:v)cout<<messi<<" "; cout<<"\n"
using namespace std;
typedef int ll;
typedef pair<ll,ll> ii;
#define ask are_connected

ll n,d;

ll find(vector<ll>&a,vector<ll>&b){
	vector<ll>pre;
	fore(i,0,SZ(a)){
		pre.pb(a[i]);
		if(ask(b,pre))return i;
	}
	/*ll l=0,r=SZ(a)-1;
	while(l<=r){
		ll m=(l+r)/2;
		vector<ll>pre;
		fore(i,0,m+1)pre.pb(a[i]);
		if(ask(b,pre))r=m-1;
		else l=m+1;
	}
	return l;*/
}

vector<ll> rot(vector<ll>a, ll d){ // <--
	ll n=SZ(a);
	vector<ll>ai(n);
	fore(i,0,n)ai[i]=a[(i+d)%n];
	return ai;
}

vector<int> longest_trip(int N, int D){
	n=N,d=D;
	//vector<ll>tr={0,1,2,3,4,5};
	//imp(tr); imp(rot(tr,2));
	vector<vector<ll>>h;
	h.pb({0});
	h.pb({1});
	fore(x,2,n){
		if(ask({h[0].back()},{x}))h[0].pb(x);
		else if(ask({h[1].back()},{x}))h[1].pb(x);
		else {
			while(SZ(h[1]))h[0].pb(h[1].back()),h[1].pop_back();
			h[1]={x};
		}
	}
	/*if(SZ(h[0])>SZ(h[1]))return h[0];
	else return h[1];*/
	//imp(h[0]); imp(h[1]);
	if(ask({h[0].back()},{h[1][0]}));//cout<<"db0\n";
	else if(ask({h[0].back()},{h[1].back()}))reverse(ALL(h[1]));//,cout<<"db1\n";
	else if(ask({h[0][0]},{h[1][0]}))reverse(ALL(h[0]));//,cout<<"db2\n";
	else if(ask({h[0][0]},{h[1].back()}))
		reverse(ALL(h[0])),reverse(ALL(h[1]));//,cout<<"db3\n";
	else {
		if(!ask(h[0],h[1])){
			//cout<<"db4\n";
			if(SZ(h[0])>SZ(h[1]))return h[0];
			else return h[1];
		}
		//assert(0);
		/*cout<<"DEBUG\n";
		imp(h[0]); imp(h[1]);*/
		
		ll x=find(h[0],h[1]);
		vector<ll>fi={h[0][x]};
		ll y=find(h[1],fi);
		//assert(ask({h[0][x]}))
		//cout<<x<<" "<<y<<"\n";
		h[0]=rot(h[0],x+1),h[1]=rot(h[1],y);
		//imp(h[0]); imp(h[1]);
		//cout<<"END\n";
		//fore(i,0,SZ(h[0])-1)
	}
	for(auto i:h[1])h[0].pb(i);
	h[1].clear();
	return h[0];
}

Compilation message (stderr)

longesttrip.cpp: In function 'll find(std::vector<int>&, std::vector<int>&)':
longesttrip.cpp:20:12: warning: control reaches end of non-void function [-Wreturn-type]
   20 |  vector<ll>pre;
      |            ^~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...