Submission #314824

#TimeUsernameProblemLanguageResultExecution timeMemory
314824kostia244Stations (IOI20_stations)C++17
76 / 100
1075 ms1120 KiB
#include "stations.h"
#include <bits/stdc++.h>
#define all(x) begin(x), end(x)
using namespace std;
namespace lab {
vector<vector<int>> g;
vector<int> tin, tout, d;
int timer;
void dfs(int v, int p) {
	tin[v] = ++timer;
	for(auto &i : g[v]) if(i != p) {
		d[i] = d[v]+1;
		dfs(i, v);
	}
	tout[v] = ++timer;
}
};
std::vector<int> label(int n, int k, std::vector<int> u, std::vector<int> v) {
	using namespace lab;
	timer = -1;
	tin = tout = d = vector<int>(n);
	g.assign(n, vector<int>());
	for(int i = 0; i+1 < n; i++) {
		g[u[i]].push_back(v[i]);
		g[v[i]].push_back(u[i]);
	}
	dfs(0, 0);
	std::vector<int> labels(n);
	for (int i = 0; i < n; i++) {
		labels[i] = (d[i]&1) ? tout[i] : tin[i];
	}
	return labels;
}
int rv(int a) { return a; }
int find_next_station(int s, int t, std::vector<int> c) {
	int type, par;
	type = s > *max_element(all(c));//tout
	if(s) {
		if(type == 0) {//tin here, tout in neighbors
			par = *max_element(all(c));
			c.erase(max_element(all(c)));
		} else {//tout here
			par = *min_element(all(c));
			c.erase(min_element(all(c)));
		}
	} else par = -1;
	if(!type) {
		if(rv(t) < rv(s)) return par;
		int mn = 1<<30;
		for(auto &i : c) if(rv(t) <= rv(i)) mn = min(mn, i);
		if(mn == 1<<30) mn = par;
		return mn;
	}
	if(t > s) return par;
	int mx = -1;
	for(auto &i : c) if(rv(t) >= rv(i)) mx = max(mx, i);
	if(mx == -1) mx = par;
	return mx;
}
#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...