| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 1311389 | hashim_bas | 기지국 (IOI20_stations) | C++20 | 0 ms | 0 KiB |
#pragma GCC optimize("O3")
#pragma GCC optimize("Ofast")
#include <bits/stdc++.h>
// #include "stations.h"
using namespace std;
#define int long long
#define ll long long
#define ld long double
#define fi first
#define se second
#define mo 1000000007
#define pb push_back
#define all(x) x.begin(),x.end()
#define iter(i,st) auto i=st.begin(); i!=st.end();i++
int cntt=-1 ;
void dfs(int nd,int pr,vector<vector<int> > &adj,vector<int> &lab,int dpth){
if(!(dpth%2))lab[nd]=cntt++ ;
for(auto &i : adj[nd]){
if(i==pr)continue ;
dfs(i,nd,adj,lab,dpth+1) ;
}
if(dpth%2)lab[nd]=cntt++ ;
return ;
}
vector<int> label(int n, int k, vector<int> u, vector<int> v) {
vector<int> labels(n);
vector<vector<int> > adj(n) ;
for (int i = 0; i < n; i++) {
adj[u[i]].pb(v[i]) ;
adj[v[i]].pb(u[i]) ;
}
dfs(0,0,adj,labels,0) ;
return labels;
}
int find_next_station(int s, int t, vector<int> c) {
sort(all(c)) ;
int op,cl ;
int n=c.size() ;
if(n==1){
return c[0] ;
}
bool b=0 ;
if(s==0){
op=0 ;
cl=c[n-1]+1 ;
}
else if(c[0]>s){
b=1 ;
op=s ;
cl=c[n-2]+1 ;
}
else if(c[0]<s){
cl[s] ;
op=c[1]-1 ;
}
if(t>cl||t<op){
if(b)return c[n-1] ;
return c[0] ;
}
if(b){
for(int i=0;i<n-1;i++){
if(c[i]>=t)return c[i] ;
}
}
else{
for(int i=n-1;i>0;i--){
if(c[i]<=t)return c[i] ;
}
}
}
