Submission #653254

#TimeUsernameProblemLanguageResultExecution timeMemory
653254mychecksedadStations (IOI20_stations)C++17
0 / 100
806 ms680 KiB
#include <bits/stdc++.h>
using namespace std;
#define pb push_back


vector<int> label(int n, int k, vector<int> u, vector<int> v){
	int timer = 0, tin[n], tout[n], dep[n];
	vector<vector<int>> g(n);
	for(int i = 0; i < n-1; ++i) g[u[i]].pb(v[i]), g[v[i]].pb(u[i]);
	function<void(int, int)> dfs = [&](int v, int p){
		tin[v] = timer++;
		dep[v] = dep[p] + 1;
		for(int u: g[v]){
			if(u != p) dfs(u, v);
		}
		tout[v] = timer++;
	};
	dep[0] = 0;
	dfs(0, 0);
	vector<int> l;
	for(int i = 0; i < n; ++i) l.pb(dep[i]%2 ? tin[i] : tout[i]);
	// sort(l.begin(), l.end());
	// for(int i = 0; i < n; ++i) l[i] = lower_bound(l.begin(), l.end(), l[i]) - l.begin();
	return l;
}
int find_next_station(int s, int t, vector<int> c){
	for(int p: c){
		if(p==t) return t;
	}
	if(s == 0){
		for(int i = 0; i < c.size(); ++i){
			if(c[i] >= t){
				return c[i];
			}
		}
	}
	if(s <= c[0]){
		int par = c.back();	
		if(t <= s || t >= par)
			return par;
		for(int i = 0; i < c.size() - 1; ++i){
			if(c[i] >= t){
				return c[i];
			}
		}
	}
	int par = c[0];
	if(t >= s || t <= par){
		return par;
	}
	for(int i = c.size() - 1; i >= 1; --i){
		if(c[i] <= t){
			return c[i];
		}
	}
	return -1;
}

Compilation message (stderr)

stations.cpp: In function 'int find_next_station(int, int, std::vector<int>)':
stations.cpp:31:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   31 |   for(int i = 0; i < c.size(); ++i){
      |                  ~~^~~~~~~~~~
stations.cpp:41:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   41 |   for(int i = 0; i < c.size() - 1; ++i){
      |                  ~~^~~~~~~~~~~~~~
#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...