Submission #802147

#	Time	Username	Problem	Language	Result	Execution time	Memory
802147		I_Love_EliskaM_	Stations (IOI20_stations)	C++14	10 / 100	751 ms	640 KiB

stations

#include "stations.h"
#include <bits/stdc++.h>
using namespace std;
#define forn(i,n) for(int i=0;i<n;++i)
#define pb push_back
#define all(x) x.begin(), x.end()
using ll = long long;

const int N=1e3;
int l[N], r[N];
vector<int> adj[N];
int nxt=0;
void dfs(int u, int p) {
	l[u]=r[u]=nxt++;
	for(auto&v:adj[u]) {
		if (v==p) continue;
		dfs(v,u);
		r[u]=max(r[u],r[v]);
	}
}

vector<int> label(int n, int k, vector<int>u, vector<int> v) {
	forn(i,n) adj[i].clear();
	forn(i,n-1) {
		adj[u[i]].pb(v[i]);
		adj[v[i]].pb(u[i]);
	}
	dfs(0,-1);
	vector<int> ans(n);
	forn(i,n) ans[i]=l[i]+N*r[i];
	return ans;
}

int find_next_station(int s, int t, vector<int> adj) {
	int l=s%N, r=s/N;
	int lx=t%N, rx=t/N;
	if (l<=lx && rx<=r) {
		for(auto&x:adj) {
			int lq=x%N, rq=x/N;
			if (lq<l) continue;
			if (lq<=lx && rx<=rq) return x;
		}
	} else {
		for(auto&x:adj) {
			int lq=x%N, rq=x/N;
			if (lq<l) return x;
		}
	}
	//assert(0);
	exit(1);
}

Compilation message (stderr)

stations.cpp: In function 'int find_next_station(int, int, std::vector<int>)':
stations.cpp:45:16: warning: unused variable 'rq' [-Wunused-variable]
   45 |    int lq=x%N, rq=x/N;
      |                ^~

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