比赛 20160415x 评测结果 AAAAAAEEAE
题目名称 游戏内测 最终得分 70
用户昵称 Aglove 运行时间 1.510 s
代码语言 C++ 内存使用 21.29 MiB
提交时间 2016-04-15 15:55:37
显示代码纯文本
#include<cstdio>
#include<cstring>
#include<iostream>
#include<cstdlib>
#include<algorithm>
#include<vector>
using namespace std;

const int maxn=500010;
int n,cnt,u,v;
int h[maxn];
struct edge{
	int to,next;
}G[maxn<<1];
int d[maxn],f[maxn],c[maxn];
vector<int>son[maxn];
bool cmp(const int &a,const int &b){
	return max(f[a],d[a]+2+f[b])<max(f[b],d[b]+2+f[a]);
}
void read(int &num){
	num=0;char ch=getchar();
	while(ch<'!')ch=getchar();
	while(ch>='0'&&ch<='9')num=num*10+ch-'0',ch=getchar();
}
void add(int x,int y){++cnt;G[cnt].to=y;G[cnt].next=h[x];h[x]=cnt;}
void DFS(int u,int fa){
	for(int i=h[u];i;i=G[i].next){
		int v=G[i].to;
		if(v==fa)continue;
		son[u].push_back(v);
		DFS(v,u);
		d[u]=d[u]+d[v]+2;
	}
	int sz=son[u].size();
	if(!sz){d[u]=0;f[u]=c[u];return;}
	sort(son[u].begin(),son[u].end(),cmp);
	int sum=0;
	for(int i=0;i<sz;++i){
		f[u]=max(f[u],sum+f[son[u][i]]+1);
		sum+=d[son[u][i]]+2;
	}f[u]=max(f[u],c[u]);
}
int main(){
	freopen("gamebeta.in","r",stdin);
	freopen("gamebeta.out","w",stdout);
	read(n);
	for(int i=1;i<=n;++i)read(c[i]);
	for(int i=1;i<n;++i){
		read(u);read(v);
		add(u,v);add(v,u);
	}
	DFS(1,-1);
	printf("%d\n",max(f[1],d[1]+c[1]));
	return 0;
}