记录编号	317340	评测结果	AAAAAAAAAA
题目名称	[HZOI 2016] 简单的最近公共祖先	最终得分	100
用户昵称	AntiLeaf	是否通过	通过
代码语言	C++	运行时间	2.258 s
提交时间	2016-10-07 21:12:56	内存使用	30.81 MiB

#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
using namespace std;
const int maxn=1000010;
namespace mine{
	template<class T>inline void readint(T &__x){
		static int __c;
		//static bool __neg;
		__x=0;
		//__neg=false;
		do __c=getchar();while(__c==' '||__c=='\n'||__c=='\r'||__c=='\t');
		/*if(__c=='-'){
			__neg=true;
			__c=getchar();
		}*/
		for(;__c>='0'&&__c<='9';__c=getchar())__x=__x*10+(__c^48);
		//if(__neg)__x=-__x;
	}
	template<class T>inline void putint(T __x){
		static int __a[40],__i,__j;
		//static bool __neg;
		//__neg=__x<0;
		//if(__neg)__x=-__x;
		__i=0;
		do{
			__a[__i++]=__x%(T)10^(T)48;
			__x/=10;
		}while(__x);
		//if(__neg)putchar('-');
		for(__j=__i-1;__j^-1;__j--)putchar(__a[__j]);
	}
}
using namespace mine;
struct edge{
	int to,prev;
}e[maxn<<1];
void insert(int,int);
void dfs(int);
LL ans=0ll;
int n,len=0,last[maxn]={0},w[maxn],size[maxn],prt[maxn]={0},x,y;
int main(){
#define MINE
#ifdef MINE
	freopen("easy_LCA.in","r",stdin);
	freopen("easy_LCA.out","w",stdout);
#endif
	readint(n);
	for(int i=1;i<=n;i++)readint(w[i]);
	for(int i=1;i<n;i++){
		readint(x);
		readint(y);
		insert(x,y);
		insert(y,x);
	}
	dfs(1);
	putint(ans);
#ifndef MINE
	printf("\n--------------------DONE--------------------\n");
	for(;;);
#endif
	return 0;
}
void insert(int x,int y){
	e[++len].to=y;
	e[len].prev=last[x];
	last[x]=len;
}
inline void dfs(int x){
	size[x]=1;
	for(int i=last[x];i;i=e[i].prev){
		if(e[i].to==prt[x])continue;
		prt[e[i].to]=x;
		dfs(e[i].to);
		ans+=(LL)size[x]*(LL)size[e[i].to]*(LL)w[x];
		size[x]+=size[e[i].to];
	}
	ans+=w[x];
}
/*
3
1 2 3
1 2
1 3
Answer:
9
*/