Submission #336977

#	Time	Username	Problem	Language	Result	Execution time	Memory
336977		Register	Stations (IOI20_stations)	C++14	100 / 100	1178 ms	1248 KiB

stations

#include <bits/stdc++.h>
#define y e[x][i]
using namespace std;
typedef vector<int> vec;
void dfs(int x,int fa,bool d,int&sum,const vector<vec> &e,vec&res){
	if(d) res[x]=sum++;
	for(int i=0;i<e[x].size();i++)
		if(y!=fa) dfs(y,x,d^1,sum,e,res);
	if(!d) res[x]=sum++;
}
vec label(int n,int k,vec u,vec v){
	vec res(n);vector<vec> e(n,vec());
	for(int i=0;i<n-1;i++) {e[u[i]].push_back(v[i]);e[v[i]].push_back(u[i]);}
	int sum=0;dfs(0,-1,1,sum,e,res);return res;
}
int find_next_station(int s,int t,vec c){
	if(c[0]>s){
		if(t<s||c.back()<t) return c.back();
		return *lower_bound(c.begin(),c.end(),t);
	}
	if(t<c[0]||s<t) return c[0];
	return *(upper_bound(c.begin(),c.end(),t)-1);
}

Compilation message (stderr)

stations.cpp: In function 'void dfs(int, int, bool, int&, const std::vector<std::vector<int> >&, vec&)':
stations.cpp:7:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    7 |  for(int i=0;i<e[x].size();i++)
      |              ~^~~~~~~~~~~~

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