Submission #377834

#TimeUsernameProblemLanguageResultExecution timeMemory
377834autumn_eelStations (IOI20_stations)C++14
36.32 / 100
1063 ms1196 KiB
#include "stations.h"
#include <bits/stdc++.h>
#define rep(i,n)for(int i=0;i<int(n);i++)
using namespace std;
typedef long long ll;

static vector<vector<int>>E;
static vector<int>labels;
static int cnt=0;

void dfs(int v,int p){
	int L=cnt++;
	for(int u:E[v]){
		if(u==p)continue;
		dfs(u,v);
	}
	int R=cnt;
	labels[v]=L*1001+R;
}

std::vector<int> label(int n, int k, std::vector<int> u, std::vector<int> v){
	E=vector<vector<int>>(n);
	labels=vector<int>(n);
	cnt=0;
	rep(i,n-1){
		E[u[i]].push_back(v[i]);
		E[v[i]].push_back(u[i]);
	}
	dfs(0,-1);
	return labels;
}

int find_next_station(int s, int t, std::vector<int> c) {
	int sl=s/1001,sr=s%1001;
	int tl=t/1001,tr=t%1001;
	for(int u:c){
		int ul=u/1001,ur=u%1001;
		if(ul<=sl&&sr<=ur)continue;
		if(ul<=tl&&tr<=ur)return u;
	}
	for(int u:c){
		int ul=u/1001,ur=u%1001;
		if(ul<=sl&&sr<=ur)return u;
	}
	
	return -1;
}
#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...