【noip2014】联合权值

【noip2014】联合权值

题目描述 Description

输入描述 Input Description

输出描述 Output Description

样例输入 Sample Input

样例输出 Sample Output

数据范围及提示 Data Size & Hint

不一定算是树形dp，也就是个暴力= =

一个点周围所有相邻的点之间的距离一定是2，存下来一个点周围所有点权值的最大值和次大值然后就得到了ans1

权值之和就是两两乘起来，存下来一个点周围所有点权值的和，然后运用一下结合律就能算出ans2

【代码】

#include<iostream>
#include<cstring>
#include<cstdio>
#define p 10007
using namespace std;
struct hp{
	long long u,v;
}e[200005];
long long n,ans1,ans2,u,v;
long long s[200005],w[200005],max1[200005],max2[200005];
//s[i]存储i点周围所有点的权值，w[i]存储i点的权值,maxn1[i]存放i点周围权值的最大值，max2[i]存储i点周围权值的次大值 
int main(){
	scanf("%lld",&n);
	for (int i=1;i<n;++i)
	  scanf("%lld%lld",&e[i].u,&e[i].v);
	for (int i=1;i<=n;++i)
	  scanf("%lld",&w[i]);
	for (int i=1;i<n;++i){
		u=e[i].u; v=e[i].v;
		s[u]+=w[v];
		s[v]+=w[u];
		if (w[u]>max1[v]) {max2[v]=max1[v]; max1[v]=w[u];}
		else if (w[u]>max2[v]&&w[u]<=max1[v]) max2[v]=w[u]; 
		if (w[v]>max1[u]) {max2[u]=max1[u]; max1[u]=w[v];}
		else if (w[v]>max2[u]&&w[v]<=max1[u]) max2[u]=w[v]; 
	}
	for (int i=1;i<=n;++i)
	  ans1=max(ans1,max1[i]*max2[i]);
	for (int i=1;i<n;++i){
		u=e[i].u; v=e[i].v;
		ans2=(ans2+w[v]*(s[u]-w[v])%p)%p;
		ans2=(ans2+w[u]*(s[v]-w[u])%p)%p;
	}
	printf("%lld %lld",ans1,ans2);
}