# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
1241254 | hssaan_arif | Stations (IOI20_stations) | C++20 | 0 ms | 0 KiB |
#include "stations.h"
#include <vector>
#include <bits/stdc++.h>
using namespace std;
#define int long long
#define endl "\n"
#define pb push_back
const int N = 1e3 + 5;
int a[N] , now;
vector<int> lis[N];
void dfs(int u , int par , int di){
if ((di & 1) == 0){
a[u] = now++;
}
for (int v : lis[u]){
if (v != par){
dfs(v , u , di+1);
}
}
if ((di & 1) == 1){
a[u] = now++;
}
}
vector<int> label(int n, int k, std::vector<int> u, std::vector<int> v) {
vector<int> labels(n);
for (int i = 1 ; i < n ; i++){
lis[u[i]].pb(v[i]);
lis[v[i]].pb(u[i]);
}
dfs(1 , 1 , 0);
for (int i = 0; i < n; i++) {
labels[i] = a[i];
}
return labels;
}
int find_next_station(int s, int t, std::vector<int> c) {
int n = c.size();
if (c.back() < s){
if (s < t){
return c[0];
}else{
int pr = 0;
for (int i = 1 ; i < n ; i++){
if (c[i] > t){
break;
}else{
pr++;
}
}
return c[pr];
}
}else{
int pr = 0;
for (int i = 1 ; i < n ; i++){
if (t > c[i]){
break;
}else{
pr++;
}
}
return c[pr];
}
}