Submission #1052380

#	Time	Username	Problem	Language	Result	Execution time	Memory
1052380		fuad27	Stations (IOI20_stations)	C++17	76 / 100	506 ms	1196 KiB

stations

#include "stations.h"
#include <bits/stdc++.h>
using namespace std;
const int N = 1010;
vector<int> g[N];
int tin[N], tout[N], dp[N];
int tim=0, tim2=0;
void dfs(int at, int p) {
//	if(dp[at])
	tin[at]=tim++;
	for(int to:g[at]) 
	{
		if(to==p)continue;
		dp[to] = dp[at]^1;
		dfs(to, at);
	}
//	if(!dp[at])
	tout[at]=tim++;
}
std::vector<int> label(int n, int k, std::vector<int> u, std::vector<int> v) {
	std::vector<int> labels(n);
	for(int i = 0;i<=n;i++)g[i].clear();
	tim=0;
	for(int i = 0;i<n-1;i++) {
		g[u[i]].push_back(v[i]);
		g[v[i]].push_back(u[i]);
	}
	dfs(0,0);
	for (int i = 0; i < n; i++) {
		if(dp[i])labels[i] = tin[i];
		else labels[i] = tout[i];
//		cout << i << " " << labels[i] << endl;
	}
	return labels;
}
 
int find_next_station(int s, int t, std::vector<int> c) {
	vector<int> kk = c;
	if(c.size()==1)return c[0];
	sort(kk.begin(), kk.end());
	if(kk[0] >= s) {
		if(t <= s)return kk.back();
		for(int i = 0;i<kk.size();i++) {
			if(kk[i] >= t)return kk[i];
		}
		return kk.back();
	}
	else if(kk.back() <= s) {
		if(t >= s)return kk[0];
		for(int i = kk.size()-1;i>=0;i--) {
			if(kk[i] <= t) {
				return kk[i];
			}
		}
		return kk[0];
	}
	return 0;
}

Compilation message (stderr)

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

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