Submission #939832

#	Time	Username	Problem	Language	Result	Execution time	Memory
939832		weajink	Longest Trip (IOI23_longesttrip)	C++17	100 / 100	11 ms	860 KiB

longesttrip

#include <bits/stdc++.h>
#include "longesttrip.h"
using namespace std;
typedef long long ll;

vector<int> sciezka[2];

void dodaj_wierzcholek(int v){
	if (!sciezka[0].size() || are_connected({v},{sciezka[0].back()})){
		sciezka[0].push_back(v);
		return;
	}
	if (!sciezka[1].size() || are_connected({v},{sciezka[1].back()})){
		sciezka[1].push_back(v);
		return;
	}
	for (int i = (int)sciezka[1].size()-1; i >= 0; i--) sciezka[0].push_back(sciezka[1][i]);
	sciezka[1] = {v};
}
void dodaj_wierzcholki(int v1, int v2){
	if (are_connected({v1},{v2})){
		dodaj_wierzcholek(v1);
		if (sciezka[0].back() == v1) sciezka[0].push_back(v2);
		else sciezka[1].push_back(v2);
		return;
	}
	if (!are_connected({v1},{sciezka[0].back()})) swap(v1,v2);
	sciezka[0].push_back(v1);
	if (are_connected({v1},{sciezka[1].back()})){
		reverse(sciezka[1].begin(),sciezka[1].end());
		for (int x : sciezka[1]) sciezka[0].push_back(x);
		sciezka[1] = {v2};
	}else sciezka[1].push_back(v2);
}
int zbinsearchuj(vector<int> A, vector<int> B){
	if (A.size() == 1) return A.back();
	vector<int> polowy[2];
	for (int i = 0; i < (int)A.size(); i++){
		polowy[i>=((int)A.size()/2)].push_back(A[i]);
	}
	if (are_connected(polowy[0],B)) return zbinsearchuj(polowy[0],B);
	else return zbinsearchuj(polowy[1],B); 
}	
vector<int> longest_trip(int N, int D){
	sciezka[0].clear();
	sciezka[1].clear();
	for (int i = 0; i < N; i++){
		if (sciezka[0].size() && sciezka[1].size() && i < N-1){
			dodaj_wierzcholki(i,i+1);
			i++;
		}else dodaj_wierzcholek(i);
	}
	if (sciezka[0].size() < sciezka[1].size()) swap(sciezka[0],sciezka[1]);
	if (!sciezka[1].size()) return sciezka[0];
	if (!are_connected(sciezka[0],sciezka[1])) return sciezka[0];
	//
	if (are_connected({sciezka[0].back()},{sciezka[1][0]})){
		vector<int> wynik;
		for (int x : sciezka[0]) wynik.push_back(x);
		for (int x : sciezka[1]) wynik.push_back(x);
		return wynik;
	}
	reverse(sciezka[0].begin(),sciezka[0].end());
	if (are_connected({sciezka[0].back()},{sciezka[1][0]})){
		vector<int> wynik;
		for (int x : sciezka[0]) wynik.push_back(x);
		for (int x : sciezka[1]) wynik.push_back(x);
		return wynik;
	}
	reverse(sciezka[1].begin(),sciezka[1].end());
	if (are_connected({sciezka[0].back()},{sciezka[1][0]})){
		vector<int> wynik;
		for (int x : sciezka[0]) wynik.push_back(x);
		for (int x : sciezka[1]) wynik.push_back(x);
		return wynik;
	}
	reverse(sciezka[0].begin(),sciezka[0].end());
	if (are_connected({sciezka[0].back()},{sciezka[1][0]})){
		vector<int> wynik;
		for (int x : sciezka[0]) wynik.push_back(x);
		for (int x : sciezka[1]) wynik.push_back(x);
		return wynik;
	}
	//
	int a = zbinsearchuj(sciezka[0],sciezka[1]);
	int b = zbinsearchuj(sciezka[1],{a});
	vector<int> wy[2];
	bool flag = 0;
	for (int x : sciezka[0]){
		wy[flag].push_back(x);
		if (x == a) flag = 1;
	}
	vector<int> wynik;
	for (int x : wy[1]) wynik.push_back(x);
	for (int x : wy[0]) wynik.push_back(x);
	wy[0] = wy[1] = {};	
	flag = 0;
	for (int x : sciezka[1]){
		if (x == b) flag = 1;
		wy[flag].push_back(x);
	}
	for (int x : wy[1]) wynik.push_back(x);
	for (int x : wy[0]) wynik.push_back(x);
	return wynik;
}

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