记录编号 337266 评测结果 AAAAAAAAAA
题目名称 [NOIP 2014]联合权值 最终得分 100
用户昵称 GravatarRapiz 是否通过 通过
代码语言 C++ 运行时间 0.407 s
提交时间 2016-11-04 08:59:51 内存使用 5.68 MiB
显示代码纯文本
#include<cstdio>
#include<vector>
#include<algorithm>
#define file(x) "linkb."#x
typedef long long ll;
using std::max;
const int V=200010,L=1<<15,M=10007;
namespace I{
	char buf[L],*s,*t;
	char gc(){
		if(s==t) t=(s=buf)+fread(buf,1,L,stdin);
		if(s==t) return EOF;
		return *s++;
	}
	int gi(){
		int s=0,ch=gc();
		while(!(ch>='0'&&ch<='9')) ch=gc();
		while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=gc();
		return s;
	}
}using I::gi;
std::vector<int> to[V];
int n,w[V],mxw[V][2],sum[V],wsum,wmax;
inline ll neg(ll a){
	if(a>=0) return a;
	else return a+(-a+M-1)/M*M;
}
void dfs(int u,int fa){
	int v;
	for(int i=0;i<to[u].size();i++) if((v=to[u][i])!=fa){
		dfs(v,u);
		sum[u]=(sum[u]+w[v])%M;
		if(w[v]>mxw[u][0]) mxw[u][1]=mxw[u][0],mxw[u][0]=w[v];
		else if(w[v]>mxw[u][1]) mxw[u][1]=w[v];
	}
	for(int i=0;i<to[u].size();i++) if((v=to[u][i])!=fa)
		wsum=(wsum+neg(sum[u]-w[v])*w[v])%M;
	wsum=(wsum+(ll)sum[u]*w[fa]*2)%M;
	wmax=max(wmax,max(mxw[u][0]*mxw[u][1],mxw[u][0]*w[fa]));
}
int main(){
	freopen(file(in),"r",stdin);
	freopen(file(out),"w",stdout);
	n=gi();
	for(int i=1;i<n;i++){
		int u,v;u=gi(),v=gi();
		to[u].push_back(v);
		to[v].push_back(u);
	}
	for(int i=1;i<=n;i++) w[i]=gi();
	dfs(1,0);
	printf("%d %d",wmax,wsum);
}