比赛 20160415x 评测结果 AAAAWWWAWW
题目名称 游戏内测 最终得分 50
用户昵称 前鬼后鬼的守护 运行时间 1.366 s
代码语言 C++ 内存使用 21.78 MiB
提交时间 2016-04-15 16:12:58
显示代码纯文本
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<ctime>
#include<cmath>
#include<algorithm>
#define FILE "gamebeta"
typedef long long ll;
typedef double lf;
namespace IO{
	char buf[1<<15],*fs,*ft;
	inline char getc(){return (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<15,stdin),fs==ft))?0:*fs++;}
	inline int read(){
		int x=0,rev=0,ch=getc();
		while(ch<'0'||ch>'9'){if(ch=='-')rev=1;ch=getc();}
		while(ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+ch-'0';ch=getc();}
		return rev?-x:x;
	}
}using namespace IO;
const int MAXN(500000),D(50),INF(1<<30);
int f[MAXN+D];
int n,c[MAXN+D];
namespace Tree{
	struct edge{
		int y,next;
		static int tot;
	}e[MAXN*2+D];
	int head[MAXN+D],edge::tot;
	inline void connect(int x,int y){
		int& k=edge::tot;
		e[++k].next=head[x];e[head[x]=k].y=y;
		e[++k].next=head[y];e[head[y]=k].y=x;
	}
	inline int max(int x,int y){return x>y?x:y;}
	inline bool by_time(const int& x,const int& y){return f[x]>f[y];}
	void dp(){
		static int par[MAXN+D],size[MAXN+D];
		static int stack[MAXN+D],top=0;
		static bool oped[MAXN+D];
		static int son[MAXN+D];
		oped[stack[++top]=1]=false;
		while(top){
			int x=stack[top];
			if(oped[x]){
				size[par[x]]+=size[x];
				int tot=0;
				for(int i=head[x],y;i;i=e[i].next)
					if((y=e[i].y)!=par[x])
						son[++tot]=y;
				std::sort(son+1,son+tot+1,by_time);
				f[x]=c[x];
				for(int i=1,cur=0;i<=tot;i++){
					f[x]=max(f[x],1+cur+f[son[i]]);
					cur+=size[son[i]]<<1;
				}
				top--;
			}
			else{
				oped[x]=true;
				size[x]=1;
				for(int i=head[x],y;i;i=e[i].next)
					if((y=e[i].y)!=par[x])
						par[y]=x,oped[stack[++top]=y]=false;
			}
		}
		printf("%d\n",max(f[1],c[1]+(n-1<<1)));
	}
}
void init(){
	n=read();
	for(int i=1;i<=n;i++)
		c[i]=read();
	for(int i=1;i<n;i++)
		Tree::connect(read(),read());
}
int main(){
	freopen(FILE".in","r",stdin);
	freopen(FILE".out","w",stdout);
	init();
	Tree::dp();
	return 0;
}