PiPi 家族

PiPi 家族
$仔细想想,好像排序挺管用的，所以就对depth[s]+a[s]排序，因为可以发现若v\in son[s],其中son[s]表示s的子树，$
$若有a[s]+depth[s]>a[v]+depth[v],因为子树要先被访问，所以v一定不会为答案做出贡献，所以可以将v合并到s中$

$仔细思考，我们现在得到一颗从上到下递增的树，我们发现大搭配小，小配大就做完了，（感性理解）$

$简单证明一下（为了防止被打），因为a[s]+depth[s]<[v]+depth[v],所以小时间一定会先配最下面（所以我们可以只比较子节点的优劣），如果它的t变大，则另一个地方（不妨设为x）的t变小，则另一个地方的x的depth[x]+a[x]+t[x]一定没有depth[v]+a[v]+t[v]大，所以一定是不优的$

#include<bits/stdc++.h>
using namespace std;
const int N=3e5+5;
int n,a[N],head[N],cnt=0,depth[N],b[N],val[N];
struct edge{
	int link,v;
}q[N<<1];
void put(int x,int y){
	q[++cnt].v=y;
	q[cnt].link=head[x];
	head[x]=cnt;
}
void dfs(int s,int fa){
	b[s]=a[s]+depth[s];
	for(int i=head[s];i;i=q[i].link){
		int v=q[i].v;
		if(v==fa) continue;
		depth[v]=depth[s]+1;
		dfs(v,s);
	}
}
int head2[N],cnt2=0,sum2=0,id[N];
struct edge2{
	int link,v;
}q2[N<<1];
void put2(int x,int y){
	q2[++cnt2].v=y;
	q2[cnt2].link=head2[x];
	head2[x]=cnt2;
}
struct node{
	int vall,id,sum;
	bool operator <(const node&bb)const{
	   return sum<bb.sum;
	}
};
priority_queue <node> myline;
void dfs2(int s,int fa,int f){
	for(int i=head[s];i;i=q[i].link){
	   int v=q[i].v;
	   if(v==fa) continue;
	   if(b[f]<b[v]){
	   	 put2(f,v);
	   	 put2(v,f);
	   	 val[v]=1;id[++sum2]=v;
	   	 dfs2(v,s,v);
	   }
	   else{
	   	 val[f]++;
	   	 dfs2(v,s,f);
       }
	}
}

int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	for(int i=1;i<n;i++){
		int u,v;
		scanf("%d%d",&u,&v);
		put(u,v),put(v,u);
	}
	dfs(1,0);
	sum2=1,id[1]=1;val[1]=1;
	dfs2(1,0,1);
	for(int i=1;i<=sum2;i++){
		myline.push((node){val[id[i]],i,b[id[i]]});
	}
    int ans=0;
	for(int t=0;t<n;t++){
		node x=myline.top();
		myline.pop();
		
		int now=x.id;val[id[now]]--;
		if(val[id[now]]){
			myline.push(node{val[id[now]],now,b[id[now]]});
		}
		else{
			ans=max(ans,b[id[now]]+t);
		}
	}
	printf("%d",ans);
}