| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 337785 | Thistle | 공장들 (JOI14_factories) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "factories.h"
#include <bits/stdc++.h>
using namespace std;
using ll=long long;
#define vec vector
#define pb push_back
#define rng(i,a,b) for(int i=(a);i<(b);i++)
#define rep(i,n) rng(i,0,n)
#define siz(a) a.size()
template<typename T>
void chmin(T& a,T b){a=min(a,b);}
template<typename T>
void chmax(T& a,T b){a=max(a,b);}
int n;
vec<int>e[600000];
ll dpt[600000];
vec<int>eul;
vec<ll>dpto;
int out[600000];
int spt[1100000][20];
int lgt[1100000];
int lca(int a,int b){
int x=out[a],y=out[b];
if(x>y) swap(x,y);
int k=lgt[(y-x+1)];
int s=spt[x][k],t=spt[y-(1<<k)+1][k];
if(dpto[s]<dpto[t]) return eul[s];
else return eul[t];
}
ll dist(int a,int b){
return dpt[a]+dpt[b]-2*dpt[lca(a,b)];
}
void Init(int N, int A[], int B[], int D[]) {
n=N;
map<pair<int, int>, ll>mp;
rep(i,n-1){
e[A[i]].pb(B[i]);
e[B[i]].pb(A[i]);
mp[make_pair(min(A[i],B[i]),max(A[i],B[i]))]=D[i];
}
auto dfs=[&](int x,int p,ll d,auto& dfs)->void{
dpt[x]=d;
out[x]=siz(eul);
eul.pb(x);dpto.pb(d);
for(auto g:e[x]){
if(g==p) continue;
ll k=mp[make_pair(min(x,g),max(x,g))];
dfs(g,x,d+k,dfs);
eul.pb(x);dpto.pb(d);
}
};
dfs(0,-1,0,dfs);
rep(i,siz(eul)) spt[i][0]=i;
rng(i,2,siz(eul)+1) lgt[i]=lgt[i>>1]+1;
rep(j,19) for(int i=0;i+(1<<(j+1))<=siz(eul);i++){
int a=spt[i][j],b=spt[i+(1<<j)][j];
if(dpto[a]<dpto[b]) spt[i][j+1]=a;
else spt[i][j+1]=b;
}
}
long long Query(int S, int X[], int T, int Y[]) {
constexpr ll inf=1e17;
ll ans=inf;
rep(i,S)rep(j,T){
chmin(ans,dist(X[i],Y[j]));
}
return ans;
}
int A[200000],B[200000],D[200000];
signed main(){
cin>>n;
rep(i,n-1){
cin>>A[i]>>B[i];D[i]=1;
A[i]--;B[i]--;
}
Init(n,A,B,D);
int q;cin>>q;
int X[1],Y[1];
rep(i,q){
cin>>X[0]>>Y[0];
X[0]--;Y[0]--;
cout<<Query(1,X,1,Y)+1<<endl;
}
}