結果
問題 | No.2634 Tree Distance 3 |
ユーザー |
👑 ![]() |
提出日時 | 2024-02-14 14:27:23 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 248 ms / 3,000 ms |
コード長 | 2,124 bytes |
コンパイル時間 | 2,288 ms |
コンパイル使用メモリ | 206,424 KB |
最終ジャッジ日時 | 2025-02-19 13:04:06 |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
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); const int B=16,C=15,len=(1<<C); vector<int> base(N*2,N); vector table(B,vector<int>(len,N)); vector<int> TL(N*2,N),TR(N*2+1,N); rep(i,0,N){ base[E[i]]=depth[i]; base[E[i+N]]=depth[i]-1; } rep(i,0,N*2){ chmin(table[0][i/B],base[i]); TL[i]=base[i]; if((i+1)%B==0){ for(int j=i;j>i+1-B;j--){ chmin(TL[j-1],TL[j]); } } else{ chmin(TR[i+1],min(TR[i],base[i])); } } rep(i,0,C-1) rep(j,0,len){ table[i+1][j]=table[i][j]; if(j+(1<<i)<len) chmin(table[i+1][j],table[i][j+(1<<i)]); } auto dist=[&](int a,int b) -> int { if(E[a]>E[b]) swap(a,b); int l=E[a]; int r=E[b]+1; int res=N; if(r-l<B){ rep(i,l,r) chmin(res,base[i]); } else{ chmin(res,TL[l]); chmin(res,TR[r]); l=l/B+1; r=r/B; int z=31-__builtin_clz(r-l); if(z>=0) chmin(res,min(table[z][l],table[z][r-(1<<z)])); } 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"; }