Submission #578547

#	Time	Username	Problem	Language	Result	Execution time	Memory
578547		Vanilla	Stations (IOI20_stations)	C++17	10 / 100	923 ms	652 KiB

stations

#include <bits/stdc++.h>
#include "stations.h"
using namespace std;
const int maxn = 1002;
vector <int> ad [maxn];
int ct = 0;

int dfs (int u, int p, vector <int> &lb) {
	lb[u] = ct++;
	int mx = 0;
	for (int v: ad[u]) {
		if (v == p) continue;
		mx = max(mx, dfs(v, u, lb));
	}
	mx = max(mx, lb[u]);
	lb[u] = (lb[u] << 9) + mx;
	return mx;
}

vector<int> label(int n, int k, vector<int> u, vector<int> v) {
	vector<int> labels(n);
	ct = 0;
	for (int i = 0; i < n; i++){
		ad[i].clear();
	}
	for (int i = 0; i < n - 1; i++){
		ad[u[i]].push_back(v[i]);
		ad[v[i]].push_back(u[i]);
	}
	dfs(0, -1, labels);
	// for (int i = 0; i < n; i++){
	// 	cout << i << " " << (labels[i] >> 10) << " " << (labels[i] & ((1 << 10) - 1)) << "\n";
	// }
	return labels;
}

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

Compilation message (stderr)

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

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