結果
問題 | No.2634 Tree Distance 3 |
ユーザー |
👑 ![]() |
提出日時 | 2024-02-15 22:05:24 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 384 ms / 3,000 ms |
コード長 | 1,650 bytes |
コンパイル時間 | 2,652 ms |
コンパイル使用メモリ | 205,484 KB |
最終ジャッジ日時 | 2025-02-19 13:18:15 |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 69 |
ソースコード
#include<bits/stdc++.h> using namespace std; #define rep(i,a,b) for(int i=int(a);i<int(b);i++) #define all(p) p.begin(),p.end() void chmin(int &a,int b){a=min(a,b);} void chmax(int &a,int b){a=max(a,b);} int main(){ ios::sync_with_stdio(false); cin.tie(nullptr); int N; cin>>N; vector<int> A(N); rep(i,0,N) cin>>A[i]; vector<vector<int>> G(N); rep(i,0,N-1){ int a,b; cin>>a>>b; a--,b--; G[a].push_back(b); G[b].push_back(a); } vector<int> ans(N); vector<int> depth(N,-1); vector<int> E(N*2); int ind=0; auto dfs=[&](auto self,int var) -> void { E[var]=ind++; for(auto x:G[var]){ if(depth[x]==-1){ depth[x]=depth[var]+1; self(self,x); } } E[var+N]=ind++; }; depth[0]=0; dfs(dfs,0); vector<int> seg(N*4); rep(i,0,N){ seg[2*N+E[i]]=depth[i]; seg[2*N+E[i+N]]=depth[i]-1; } for(int i=2*N-1;i>0;i--) seg[i]=min(seg[2*i],seg[2*i+1]); auto dist=[&](int a,int b) -> int { int res=N; int l=E[a],r=E[b]; if(l>r) swap(l,r); r++; l+=2*N; r+=2*N; while(l<r){ if(l&1) chmin(res,seg[l++]); if(r&1) chmin(res,seg[--r]); l>>=1;r>>=1; } return depth[a]+depth[b]-2*res; }; vector<int> order(N); rep(i,0,N) order[i]=i; sort(all(order),[&](int l,int r){ return A[l]>A[r]; }); int L=0,R=0; int X=order[0],Y=order[0],D=0; while(L<N){ while(R<N&&A[order[L]]==A[order[R]]){ int tmpX=dist(X,order[R]); int tmpY=dist(Y,order[R]); if(tmpX<tmpY) swap(X,Y),swap(tmpX,tmpY); if(D<tmpX) D=tmpX,Y=order[R]; R++; } rep(i,L,R){ ans[order[i]]=max(dist(order[i],X),dist(order[i],Y)); } L=R; } rep(i,0,N){ if(i) cout<<" "; cout<<ans[i]; } cout<<"\n"; }