結果

問題 No.2634 Tree Distance 3
ユーザー PersistentLifePersistentLife
提出日時 2024-02-16 21:52:42
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 223 ms / 3,000 ms
コード長 2,607 bytes
コンパイル時間 1,605 ms
コンパイル使用メモリ 178,752 KB
実行使用メモリ 98,896 KB
最終ジャッジ日時 2024-09-28 20:07:57
合計ジャッジ時間 16,762 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 209 ms
92,416 KB
testcase_01 AC 213 ms
92,420 KB
testcase_02 AC 213 ms
92,420 KB
testcase_03 AC 210 ms
92,416 KB
testcase_04 AC 208 ms
94,596 KB
testcase_05 AC 198 ms
92,424 KB
testcase_06 AC 215 ms
94,468 KB
testcase_07 AC 212 ms
94,472 KB
testcase_08 AC 166 ms
89,840 KB
testcase_09 AC 182 ms
91,988 KB
testcase_10 AC 163 ms
89,980 KB
testcase_11 AC 168 ms
92,000 KB
testcase_12 AC 162 ms
89,880 KB
testcase_13 AC 167 ms
89,996 KB
testcase_14 AC 171 ms
89,712 KB
testcase_15 AC 163 ms
89,872 KB
testcase_16 AC 154 ms
93,892 KB
testcase_17 AC 92 ms
84,020 KB
testcase_18 AC 116 ms
87,708 KB
testcase_19 AC 136 ms
92,588 KB
testcase_20 AC 52 ms
72,224 KB
testcase_21 AC 198 ms
94,412 KB
testcase_22 AC 223 ms
96,032 KB
testcase_23 AC 204 ms
98,896 KB
testcase_24 AC 197 ms
92,376 KB
testcase_25 AC 193 ms
91,528 KB
testcase_26 AC 197 ms
92,792 KB
testcase_27 AC 198 ms
91,504 KB
testcase_28 AC 195 ms
92,264 KB
testcase_29 AC 191 ms
91,684 KB
testcase_30 AC 202 ms
93,352 KB
testcase_31 AC 200 ms
93,360 KB
testcase_32 AC 197 ms
93,376 KB
testcase_33 AC 125 ms
88,244 KB
testcase_34 AC 33 ms
64,928 KB
testcase_35 AC 76 ms
82,692 KB
testcase_36 AC 47 ms
68,700 KB
testcase_37 AC 88 ms
84,120 KB
testcase_38 AC 11 ms
44,032 KB
testcase_39 AC 10 ms
41,192 KB
testcase_40 AC 10 ms
43,632 KB
testcase_41 AC 10 ms
44,432 KB
testcase_42 AC 10 ms
42,856 KB
testcase_43 AC 64 ms
77,964 KB
testcase_44 AC 45 ms
67,612 KB
testcase_45 AC 165 ms
89,784 KB
testcase_46 AC 106 ms
82,748 KB
testcase_47 AC 180 ms
90,732 KB
testcase_48 AC 198 ms
91,448 KB
testcase_49 AC 193 ms
91,452 KB
testcase_50 AC 193 ms
91,436 KB
testcase_51 AC 193 ms
91,312 KB
testcase_52 AC 205 ms
91,464 KB
testcase_53 AC 11 ms
43,296 KB
testcase_54 AC 10 ms
43,324 KB
testcase_55 AC 10 ms
43,980 KB
testcase_56 AC 10 ms
43,280 KB
testcase_57 AC 10 ms
41,136 KB
testcase_58 AC 6 ms
26,396 KB
testcase_59 AC 6 ms
27,252 KB
testcase_60 AC 203 ms
92,784 KB
testcase_61 AC 208 ms
92,792 KB
testcase_62 AC 206 ms
92,660 KB
testcase_63 AC 167 ms
90,708 KB
testcase_64 AC 110 ms
84,376 KB
testcase_65 AC 156 ms
89,520 KB
testcase_66 AC 54 ms
70,476 KB
testcase_67 AC 45 ms
69,240 KB
testcase_68 AC 167 ms
91,872 KB
testcase_69 AC 170 ms
90,692 KB
testcase_70 AC 170 ms
89,960 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

/*
Things to notice:
1. do not calculate useless values
2. do not use similar names
 
Things to check:
1. submit the correct file
2. time (it is log^2 or log)
3. memory
4. prove your naive thoughts 
5. long long
6. corner case like n=0,1,inf or n=m
7. check if there is a mistake in the ds or other tools you use
8. fileio in some oi-contest

9. module on time 
10. the number of a same divisor in a math problem
11. multi-information and queries for dp and ds problems
*/
#include<bits/stdc++.h>
using namespace std;
#define int long long
#define fi first
#define se second
#define pii pair<long long,long long>
#define mp make_pair
#define pb push_back
const int mod=998244353;
const int inf=0x3f3f3f3f;
const int INF=1e18;
vector <pii > g[200005]; 
struct LCA
{
	int dep[200005],dfn[200005],times;
	int dist[200005];
	int st[20][400005],Lg[400005];
	void dfs(int u,int fa)
	{
		dfn[u]=++times,st[0][times]=u;
		for(int i=0;i<g[u].size();i++)
		{
			int v=g[u][i].fi;
			if(v==fa) continue;
			dep[v]=dep[u]+1,dist[v]=dist[u]+1LL*g[u][i].se,dfs(v,u);
			st[0][++times]=u;
		}
	}
	int mindep(int x,int y)
	{
		if(dep[x]<dep[y]) return x;
		return y;
	}
	void build()
	{
		times=0,dep[1]=dist[1]=0;
		dfs(1,-1);
		Lg[1]=0,Lg[2]=1;
		for(int i=3;i<=times;i++) Lg[i]=Lg[i/2]+1;
		for(int k=1;k<20;k++) for(int i=1;i+(1<<k)-1<=times;i++) 
			st[k][i]=mindep(st[k-1][i],st[k-1][i+(1<<(k-1))]);
	}
	int getlca(int u,int v)
	{
		u=dfn[u],v=dfn[v];
		if(u>v) swap(u,v);
		int s=Lg[v-u+1];
		return mindep(st[s][u],st[s][v-(1<<s)+1]);
	}
	int getdis(int u,int v)
	{
		int l=getlca(u,v);
		return dist[u]+dist[v]-2LL*dist[l];	
	}
}Lca;
int n;
int a[200005],ans[200005];
pii b[200005];
void solve()
{
	cin>>n;
	for(int i=1;i<=n;i++) cin>>a[i],b[i]=mp(a[i],i);
	sort(b+1,b+1+n),reverse(b+1,b+1+n);
	for(int i=1;i<n;i++)
	{
		int u,v;
		cin>>u>>v;
		g[u].pb(mp(v,1)),g[v].pb(mp(u,1));
	}
	Lca.build();
	pii diam=mp(-1,-1);
	for(int i=1;i<=n;i++)
	{
		int j=i;
		while(j+1<=n&&b[j+1].fi==b[j].fi) j++;
		for(int l=i;l<=j;l++)
		{
			int u=b[l].se;
			if(diam==mp(-1LL,-1LL)) diam=mp(u,u);
			else
			{
				int x=diam.fi,y=diam.se;
				if(Lca.getdis(u,x)>=Lca.getdis(x,y)&&Lca.getdis(u,x)>=Lca.getdis(u,y)) diam=mp(u,x);
				else if(Lca.getdis(u,y)>=Lca.getdis(x,y)&&Lca.getdis(u,y)>=Lca.getdis(u,x)) diam=mp(u,y);
				else diam=mp(x,y);
			}
		}
		for(int l=i;l<=j;l++)
		{
			int u=b[l].se;
			ans[u]=max(Lca.getdis(u,diam.fi),Lca.getdis(u,diam.se));
		}
		i=j; 
	}
	for(int i=1;i<=n;i++) cout<<ans[i]<<" ";
}
signed main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	int _=1;
//	cin>>_;
	while(_--) solve();
	return 0;
}
0