Submission #1284564

#	Time	Username	Problem	Language	Result	Execution time	Memory
1284564		farica	Stations (IOI20_stations)	C++20	0 / 100	410 ms	580 KiB

stations

#include "stations.h"
#include <bits/stdc++.h>

using namespace std;
using vi = vector<int>;

vi labels;
vector<vi>adjL;

int dfs(int pos, int cnt, int prev=-1) {
	for(int adj: adjL[pos]) {
		if(adj == prev) continue;
		cnt = dfs(adj, cnt, pos);
	}
	labels[pos] = cnt;
	return cnt+1;
}

vector<int> label(int n, int k, std::vector<int> u, std::vector<int> v) {
	labels.resize(n);
	adjL.assign(n, vi());
	for(int i=0; i<n-1; ++i) {
		adjL[u[i]].push_back(v[i]);
		adjL[v[i]].push_back(u[i]);
	}
	dfs(0, 0);
	return labels;
}

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

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