| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 675937 | Essa2006 | 기지국 (IOI20_stations) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include<bits/stdc++.h>
using namespace std;
#define all(a) a.begin(), a.end()
vector<int>Vis, new_;
vector<vector<int>>A;
bool even;
int time_;
void dfs(int ind){
Vis[ind]=1;
bool odd=!even;
if(!odd) new_[ind]=time_++;
for(int i=0;i<A[ind].size();i++)
if(!Vis[A[ind][i]]){
even=odd;
dfs(A[ind][i]);
}
if(odd) new_[ind]=time_++;
}
vector<int> label(int n, int k, vector<int> u, vector<int> v) {
Vis.clear(), A.clear(), new_.clear();
Vis.resize(n), A.resize(n), new_.resize(n);
time_=0, even=1;
for(int i=0;i<n-1;i++)
A[u[i]].push_back(v[i]), A[v[i]].push_back(u[i]);
dfs(0);
return new_;
}
int find_next_station(int s, int t, vector<int> c) {
if(c[0]<s){
if(t>s || t==c.front()) return c.front();
for(int i=c.size()-1;i>=0;i--)
if(c[i]<=t)
return c[i];
}
else{
if(t<s || t>=c.back()) return c.back();
for(int i=0;i<c.size();i++)
if(c[i]>=t)
return c[i];
}
}
int main(){
int n, k;
cin>>n>>k;
vector<int>u(n-1), v(n-1);
for(int i=0;i<n-1;i++){
cin>>u[i]>>v[i];
}
vector<int>D=label(n, k, u, v);
int q;
cin>>q;
while(q--){
int a, b;
vector<int>c;
cin>>a>>b;
for(int i=0;i<A[a].size();i++){
c.push_back(D[A[a][i]]);
}
sort(all(c));
cout<<find_next_station(D[a], D[b], c)<<endl;
}
}