#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int Nmax=2e5+5;
int a[Nmax];
int nodes[Nmax],pos[Nmax],activ[Nmax],p[Nmax];
ll dp[Nmax];
int nodedepth[Nmax];
pair<int,int> aint[4*Nmax];
vector<int> euler;
vector<int> depths;
vector<int> v[Nmax];
int n;
int sol=0;
int sef(int x)
{
if(p[x]==x) return x;
p[x]=sef(p[x]);
return p[x];
}
bool cmp(int x,int y)
{
return a[x]<a[y];
}
void build(int nod,int l,int r)
{
if(l==r)
{
aint[nod]={depths[l],euler[l]};
return;
}
int mid=(l+r)/2;
build(2*nod,l,mid);
build(2*nod+1,mid+1,r);
aint[nod]=min(aint[2*nod],aint[2*nod+1]);
return;
}
pair<int,int> query(int nod,int l,int r,int st,int dr)
{
if(st<=l && r<=dr) return aint[nod];
int mid=(l+r)/2;
if(dr<=mid) return query(2*nod,l,mid,st,dr);
if(mid<st) return query(2*nod+1,mid+1,r,st,dr);
return min(query(2*nod,l,mid,st,mid),query(2*nod+1,mid+1,r,mid+1,dr));
}
void dfs(int nod,int prev,int depth)
{
euler.push_back(nod);
depths.push_back(depth);
nodedepth[nod]=depth;
pos[nod]=euler.size()-1;
for(auto u:v[nod])
{
if(u!=prev)
{
dfs(u,nod,depth+1);
euler.push_back(nod);
depths.push_back(depth);
}
}
}
int lca(int u,int v)
{
if(pos[u]>pos[v]) swap(u,v);
return query(1,0,euler.size()-1,pos[u],pos[v]).second;
}
void calclca()
{
dfs(1,0,0);
build(1,0,euler.size()-1);
}
int dist(int u,int v)
{
int nod=lca(u,v);
return nodedepth[u]+nodedepth[v]-2*nodedepth[nod];
}
void unite(int x,int y)
{
int sefx=sef(x);
int sefy=sef(y);
if(a[sefx]<a[sefy]) swap(sefx,sefy);
p[sefy]=sefx;
return;
}
void activate(int nod)
{
activ[nod]=1;
p[nod]=nod;
for(auto u:v[nod])
{
if(activ[u])
{
int nxt=sef(u);
dp[nod]=max(dp[nod],1LL*dist(nod,nxt)+dp[nxt]);
unite(nod,u);
}
}
}
signed main()
{
cin>>n;
int start;
for(int i=1;i<=n;i++)
{
cin>>a[i];
if(a[i]==n) start=i;
nodes[i]=i;
}
sort(nodes+1,nodes+n+1,cmp);
bool ok=0;
for(int i=1;i<n;i++)
{
int x,y;
cin>>x>>y;
v[x].push_back(y);
v[y].push_back(x);
}
calclca();
for(int i=1;i<=n;i++)
{
activate(nodes[i]);
}
cout<<dp[nodes[n]]<<'\n';
return 0;
}
Compilation message
Main.cpp: In function 'int main()':
Main.cpp:102:9: warning: variable 'start' set but not used [-Wunused-but-set-variable]
102 | int start;
| ^~~~~
Main.cpp:110:10: warning: unused variable 'ok' [-Wunused-variable]
110 | bool ok=0;
| ^~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
12888 KB |
Output is correct |
2 |
Correct |
3 ms |
12916 KB |
Output is correct |
3 |
Correct |
2 ms |
12892 KB |
Output is correct |
4 |
Correct |
2 ms |
12892 KB |
Output is correct |
5 |
Correct |
3 ms |
12892 KB |
Output is correct |
6 |
Correct |
2 ms |
12892 KB |
Output is correct |
7 |
Correct |
3 ms |
12888 KB |
Output is correct |
8 |
Correct |
3 ms |
12892 KB |
Output is correct |
9 |
Correct |
3 ms |
12892 KB |
Output is correct |
10 |
Correct |
3 ms |
13152 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
12888 KB |
Output is correct |
2 |
Correct |
3 ms |
12916 KB |
Output is correct |
3 |
Correct |
2 ms |
12892 KB |
Output is correct |
4 |
Correct |
2 ms |
12892 KB |
Output is correct |
5 |
Correct |
3 ms |
12892 KB |
Output is correct |
6 |
Correct |
2 ms |
12892 KB |
Output is correct |
7 |
Correct |
3 ms |
12888 KB |
Output is correct |
8 |
Correct |
3 ms |
12892 KB |
Output is correct |
9 |
Correct |
3 ms |
12892 KB |
Output is correct |
10 |
Correct |
3 ms |
13152 KB |
Output is correct |
11 |
Correct |
3 ms |
12892 KB |
Output is correct |
12 |
Correct |
2 ms |
12916 KB |
Output is correct |
13 |
Correct |
2 ms |
12888 KB |
Output is correct |
14 |
Correct |
2 ms |
12892 KB |
Output is correct |
15 |
Correct |
2 ms |
12892 KB |
Output is correct |
16 |
Correct |
3 ms |
12892 KB |
Output is correct |
17 |
Correct |
3 ms |
12892 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
12888 KB |
Output is correct |
2 |
Correct |
3 ms |
12916 KB |
Output is correct |
3 |
Correct |
2 ms |
12892 KB |
Output is correct |
4 |
Correct |
2 ms |
12892 KB |
Output is correct |
5 |
Correct |
3 ms |
12892 KB |
Output is correct |
6 |
Correct |
2 ms |
12892 KB |
Output is correct |
7 |
Correct |
3 ms |
12888 KB |
Output is correct |
8 |
Correct |
3 ms |
12892 KB |
Output is correct |
9 |
Correct |
3 ms |
12892 KB |
Output is correct |
10 |
Correct |
3 ms |
13152 KB |
Output is correct |
11 |
Correct |
3 ms |
12892 KB |
Output is correct |
12 |
Correct |
2 ms |
12916 KB |
Output is correct |
13 |
Correct |
2 ms |
12888 KB |
Output is correct |
14 |
Correct |
2 ms |
12892 KB |
Output is correct |
15 |
Correct |
2 ms |
12892 KB |
Output is correct |
16 |
Correct |
3 ms |
12892 KB |
Output is correct |
17 |
Correct |
3 ms |
12892 KB |
Output is correct |
18 |
Correct |
7 ms |
13404 KB |
Output is correct |
19 |
Correct |
7 ms |
13404 KB |
Output is correct |
20 |
Correct |
7 ms |
13400 KB |
Output is correct |
21 |
Correct |
7 ms |
13576 KB |
Output is correct |
22 |
Correct |
7 ms |
13404 KB |
Output is correct |
23 |
Correct |
8 ms |
13660 KB |
Output is correct |
24 |
Correct |
7 ms |
13404 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
12888 KB |
Output is correct |
2 |
Correct |
3 ms |
12916 KB |
Output is correct |
3 |
Correct |
2 ms |
12892 KB |
Output is correct |
4 |
Correct |
2 ms |
12892 KB |
Output is correct |
5 |
Correct |
3 ms |
12892 KB |
Output is correct |
6 |
Correct |
2 ms |
12892 KB |
Output is correct |
7 |
Correct |
3 ms |
12888 KB |
Output is correct |
8 |
Correct |
3 ms |
12892 KB |
Output is correct |
9 |
Correct |
3 ms |
12892 KB |
Output is correct |
10 |
Correct |
3 ms |
13152 KB |
Output is correct |
11 |
Correct |
3 ms |
12892 KB |
Output is correct |
12 |
Correct |
2 ms |
12916 KB |
Output is correct |
13 |
Correct |
2 ms |
12888 KB |
Output is correct |
14 |
Correct |
2 ms |
12892 KB |
Output is correct |
15 |
Correct |
2 ms |
12892 KB |
Output is correct |
16 |
Correct |
3 ms |
12892 KB |
Output is correct |
17 |
Correct |
3 ms |
12892 KB |
Output is correct |
18 |
Correct |
7 ms |
13404 KB |
Output is correct |
19 |
Correct |
7 ms |
13404 KB |
Output is correct |
20 |
Correct |
7 ms |
13400 KB |
Output is correct |
21 |
Correct |
7 ms |
13576 KB |
Output is correct |
22 |
Correct |
7 ms |
13404 KB |
Output is correct |
23 |
Correct |
8 ms |
13660 KB |
Output is correct |
24 |
Correct |
7 ms |
13404 KB |
Output is correct |
25 |
Correct |
2 ms |
12892 KB |
Output is correct |
26 |
Correct |
7 ms |
13404 KB |
Output is correct |
27 |
Correct |
7 ms |
13404 KB |
Output is correct |
28 |
Correct |
7 ms |
13404 KB |
Output is correct |
29 |
Correct |
7 ms |
13404 KB |
Output is correct |
30 |
Correct |
8 ms |
13300 KB |
Output is correct |
31 |
Correct |
7 ms |
13148 KB |
Output is correct |
32 |
Correct |
7 ms |
13148 KB |
Output is correct |
33 |
Correct |
7 ms |
13320 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
12888 KB |
Output is correct |
2 |
Correct |
3 ms |
12916 KB |
Output is correct |
3 |
Correct |
2 ms |
12892 KB |
Output is correct |
4 |
Correct |
2 ms |
12892 KB |
Output is correct |
5 |
Correct |
3 ms |
12892 KB |
Output is correct |
6 |
Correct |
2 ms |
12892 KB |
Output is correct |
7 |
Correct |
3 ms |
12888 KB |
Output is correct |
8 |
Correct |
3 ms |
12892 KB |
Output is correct |
9 |
Correct |
3 ms |
12892 KB |
Output is correct |
10 |
Correct |
3 ms |
13152 KB |
Output is correct |
11 |
Correct |
3 ms |
12892 KB |
Output is correct |
12 |
Correct |
2 ms |
12916 KB |
Output is correct |
13 |
Correct |
2 ms |
12888 KB |
Output is correct |
14 |
Correct |
2 ms |
12892 KB |
Output is correct |
15 |
Correct |
2 ms |
12892 KB |
Output is correct |
16 |
Correct |
3 ms |
12892 KB |
Output is correct |
17 |
Correct |
3 ms |
12892 KB |
Output is correct |
18 |
Correct |
7 ms |
13404 KB |
Output is correct |
19 |
Correct |
7 ms |
13404 KB |
Output is correct |
20 |
Correct |
7 ms |
13400 KB |
Output is correct |
21 |
Correct |
7 ms |
13576 KB |
Output is correct |
22 |
Correct |
7 ms |
13404 KB |
Output is correct |
23 |
Correct |
8 ms |
13660 KB |
Output is correct |
24 |
Correct |
7 ms |
13404 KB |
Output is correct |
25 |
Incorrect |
237 ms |
42856 KB |
Output isn't correct |
26 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
12892 KB |
Output is correct |
2 |
Correct |
3 ms |
12892 KB |
Output is correct |
3 |
Incorrect |
275 ms |
30196 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
12888 KB |
Output is correct |
2 |
Correct |
3 ms |
12916 KB |
Output is correct |
3 |
Correct |
2 ms |
12892 KB |
Output is correct |
4 |
Correct |
2 ms |
12892 KB |
Output is correct |
5 |
Correct |
3 ms |
12892 KB |
Output is correct |
6 |
Correct |
2 ms |
12892 KB |
Output is correct |
7 |
Correct |
3 ms |
12888 KB |
Output is correct |
8 |
Correct |
3 ms |
12892 KB |
Output is correct |
9 |
Correct |
3 ms |
12892 KB |
Output is correct |
10 |
Correct |
3 ms |
13152 KB |
Output is correct |
11 |
Correct |
3 ms |
12892 KB |
Output is correct |
12 |
Correct |
2 ms |
12916 KB |
Output is correct |
13 |
Correct |
2 ms |
12888 KB |
Output is correct |
14 |
Correct |
2 ms |
12892 KB |
Output is correct |
15 |
Correct |
2 ms |
12892 KB |
Output is correct |
16 |
Correct |
3 ms |
12892 KB |
Output is correct |
17 |
Correct |
3 ms |
12892 KB |
Output is correct |
18 |
Correct |
7 ms |
13404 KB |
Output is correct |
19 |
Correct |
7 ms |
13404 KB |
Output is correct |
20 |
Correct |
7 ms |
13400 KB |
Output is correct |
21 |
Correct |
7 ms |
13576 KB |
Output is correct |
22 |
Correct |
7 ms |
13404 KB |
Output is correct |
23 |
Correct |
8 ms |
13660 KB |
Output is correct |
24 |
Correct |
7 ms |
13404 KB |
Output is correct |
25 |
Correct |
2 ms |
12892 KB |
Output is correct |
26 |
Correct |
7 ms |
13404 KB |
Output is correct |
27 |
Correct |
7 ms |
13404 KB |
Output is correct |
28 |
Correct |
7 ms |
13404 KB |
Output is correct |
29 |
Correct |
7 ms |
13404 KB |
Output is correct |
30 |
Correct |
8 ms |
13300 KB |
Output is correct |
31 |
Correct |
7 ms |
13148 KB |
Output is correct |
32 |
Correct |
7 ms |
13148 KB |
Output is correct |
33 |
Correct |
7 ms |
13320 KB |
Output is correct |
34 |
Incorrect |
237 ms |
42856 KB |
Output isn't correct |
35 |
Halted |
0 ms |
0 KB |
- |