【NOIP2014提高组】联合权值

题目背景

NOIP2014提高组 Day1 试题。

题目描述

无向连通图 G 有 n 个点，n-1 条边。点从 1 到 n 依次编号，编号为 i 的点的权值为 Wi，每条边的长度均为 1。图上两点 (u, v) 的距离定义为 u 点到 v 点的最短距离。对于图 G 上的点对(u, v)，若它们的距离为 2，则它们之间会产生 Wu× Wv 的联合权值。 

请问图 G 上所有可产生联合权值的有序点对中，联合权值最大的是多少？所有联合权值之和是多少？

输入格式

第 1 行包含 1 个整数 n。 
接下来 n-1 行，每行包含 2 个用空格隔开的正整数 u、v，表示编号为 u 和编号为 v 的点之间有边相连。 
最后 1 行，包含 n 个正整数，每两个正整数之间用一个空格隔开，其中第 i 个整数表示图 G 上编号为 i 的点的权值为 Wi。

输出格式

输出共 1 行，包含 2 个整数，之间用一个空格隔开，依次为图 G 上联合权值的最大值和所有联合权值之和。由于所有联合权值之和可能很大，输出它时要对10007取余。

样例数据 1

输入

5 
1 2 
2 3 
3 4 
4 5 
1 5 2 3 10

输出

20 74

备注

【样例说明】

本例输入的图如上所示，距离为 2 的有序点对有(1,3)、(2,4)、(3,1)、(3,5)、(4,2)、(5,3)。 
其联合权值分别为 2、15、2、20、15、20。其中最大的是 20，总和为 74。 

【数据范围】 
对于 30% 的数据，1<n≤100； 
对于 60% 的数据，1<n≤2000； 
对于 100% 的数据，1<n≤200,000，0<Wi≤10,000。

 

解析：
       最初本来都想到正解了，但是由于把复杂度错误地分析成了所以没敢写只写了暴力。。。

       正解就是枚举每一个点，然后遍历从这个点出发的所有边更新答案，因为每条边最多被遍历两次，所以复杂度是级别的。。。

 

代码：

#include <bits/stdc++.h>
#define int long long
using namespace std;

const int mod=10007;
const int Max=200010;
int n,m,size,ans1,ans2;
int first[Max],w[Max],f[Max][2];
struct shu{int to,next;};
shu edge[Max<<1];

inline int get_int()
{
	int x=0,f=1;
	char c;
	for(c=getchar();(!isdigit(c))&&(c!='-');c=getchar());
	if(c=='-') f=-1,c=getchar();
	for(;isdigit(c);c=getchar()) x=(x<<3)+(x<<1)+c-'0';
	return x*f;
}

inline void build(int x,int y)
{
	edge[++size].next=first[x];
	first[x]=size;
	edge[size].to=y;
}

inline int dfs(int point)
{
	int max1=0,max2=0,sum=0;
	for(int u=first[point];u;u=edge[u].next)
	{
	  int to=edge[u].to;
	  ans2=(ans2+sum*w[to]);
	  sum+=w[to];
	  if(w[to] > max1) max2=max1,max1=w[to];
	  else max2=max(max2,w[to]);
	}
	ans1=max(ans1,max1*max2);
}

signed main()
{
	n=get_int();
	for(int i=1;i<=n-1;i++)
	{
	  int x=get_int(),y=get_int();
	  build(x,y),build(y,x);
	}
	for(int i=1;i<=n;i++) w[i]=get_int();
	for(int i=1;i<=n;i++) dfs(i);
	cout<<ans1<<" "<<ans2*2%mod<<"\n";
	return 0;
}