Submission #816635

#	Time	Username	Problem	Language	Result	Execution time	Memory
816635		tolbi	Stations (IOI20_stations)	C++17	100 / 100	865 ms	764 KiB

stations

#pragma optimize("Bismillahirrahmanirrahim")
//█▀█─█──█──█▀█─█─█
//█▄█─█──█──█▄█─█■█
//█─█─█▄─█▄─█─█─█─█
//Allahuekber
//ahmet23 orz...
//Sani buyuk Osman Pasa Plevneden cikmam diyor.
//FatihSultanMehmedHan
//YavuzSultanSelimHan
//AbdulhamidHan
#define author tolbi
#include <bits/stdc++.h>
using namespace std;
template<typename X, typename Y> istream& operator>>(istream& in, pair<X,Y> &pr) {return in>>pr.first>>pr.second;}
template<typename X, typename Y> ostream& operator<<(ostream& os, pair<X,Y> pr) {return os<<pr.first<<" "<<pr.second;}
template<typename X> istream& operator>>(istream& in, vector<X> &arr) {for(auto &it : arr) in>>it; return in;}
template<typename X> ostream& operator<<(ostream& os, vector<X> arr) {for(auto &it : arr) os<<it<<" "; return os;}
template<typename X, size_t Y> istream& operator>>(istream& in, array<X,Y> &arr) {for(auto &it : arr) in>>it; return in;}
template<typename X, size_t Y> ostream& operator<<(ostream& os, array<X,Y> arr) {for(auto &it : arr) os<<it<<" "; return os;}
#define endl '\n'
#define vint(x) vector<int> x
#define deci(x) int x;cin>>x;
#define decstr(x) string x;cin>>x;
#define cinarr(x) for (auto &it : x) cin>>it;
#define coutarr(x) for (auto &it : x) cout<<it<<" ";cout<<endl;
#define sortarr(x) sort(x.begin(),x.end())
#define sortrarr(x) sort(x.rbegin(),x.rend())
#define det(x) cout<<"NO\0YES"+x*3<<endl;
#define INF LONG_LONG_MAX
#define rev(x) reverse(x.begin(),x.end());
#define ios ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
#define tol(bi) (1LL<<((int)(bi)))
const int MOD = 1e9+7;
mt19937 ayahya(chrono::high_resolution_clock().now().time_since_epoch().count());
#include "stations.h"

std::vector<int> label(int n, int k, std::vector<int> u, std::vector<int> v) {
	vector<vector<int>> arr(n);
	vector<int> labels(n,-1);
	int ind=0;
	auto dfs = [&](int node, int lnode, bool tek, auto dfs)->void{
		ind++;
		if (tek) labels[node]=ind;
		ind++;
		for (int i = 0; i < arr[node].size(); i++){
			if (arr[node][i]==lnode) continue;
			dfs(arr[node][i],node,!tek,dfs);
		}
		ind++;
		if (!tek) labels[node]=ind;
		ind++;
	};
	for (int i = 0; i < n-1; i++){
		arr[u[i]].push_back(v[i]);
		arr[v[i]].push_back(u[i]);
	}
	dfs(0,-1,true,dfs);
	vector<pair<int,int>> sr(n);
	for (int i = 0; i < n; i++){
		sr[i]={labels[i],i};
	}
	sortarr(sr);
	for (int i = 0; i < n; i++){
		labels[sr[i].second]=i;
	}
	return labels;
}

int find_next_station(int s, int t, std::vector<int> c) {
	if (s>c.back()){
		for (int i = c.size()-1; i >= 1; i--){
			if (t>=c[i] && t<=s){
				return c[i];
			}
		}
		return c[0];
	}
	for (int i = 0; i < c.size()-1; i++){
		if (t>=s && t<=c[i]){
			return c[i];
		}
	}
	return c.back();
}

Compilation message (stderr)

stations.cpp:1: warning: ignoring '#pragma optimize ' [-Wunknown-pragmas]
    1 | #pragma optimize("Bismillahirrahmanirrahim")
      | 
stations.cpp: In function 'int find_next_station(int, int, std::vector<int>)':
stations.cpp:78:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   78 |  for (int i = 0; i < c.size()-1; i++){
      |                  ~~^~~~~~~~~~~~
stations.cpp: In instantiation of 'label(int, int, std::vector<int>, std::vector<int>)::<lambda(int, int, bool, auto:23)> [with auto:23 = label(int, int, std::vector<int>, std::vector<int>)::<lambda(int, int, bool, auto:23)>]':
stations.cpp:57:19:   required from here
stations.cpp:45:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   45 |   for (int i = 0; i < arr[node].size(); i++){
      |                   ~~^~~~~~~~~~~~~~~~~~

• Subtask 1: 5 / 5
• Subtask 2: 8 / 8
• Subtask 3: 16 / 16
• Subtask 4: 10 / 10
• Subtask 5: 61 / 61