[NOIP2014提高组]联合权值

题目：洛谷P1351、Vijos P1906、codevs3728、UOJ#16。

题目大意：有一个无向连通图，有n个点n-1条边，每个点有一个权值$W_i$，每条边长度为1。规定两个距离为2的点i和j可以产生$W_i×W_j$的联合权值。求最大的联合权值是多少，联合权值之和是多少。

解题思路：首先，距离为2的点只有两种情况：①点u和它父亲的父亲；②点u和它的兄弟。那么我们只需遍历全图，记录该点父亲的父亲即可。对于每个节点，求出它所有儿子和儿子之间的联合权值是多少，加起来即可。

这样子可能会超时。那么我们可以用一个变量记录前面儿子的权值和，然后直接乘这个和即可。最后答案乘2。

对于第一问，我们也思考以上两种情况。第①种很好考虑，第②种我们只需记录最大权值和次大权值，然后相乘就是可能的最大值了。

于是就过了。

C++ Code：

#include<cstdio>
#include<vector>
using namespace std;
vector<int>G[200002];
int n,d[200002],ans=0,Max=0;
bool b[200002]={0};
void addedge(int from,int to){
	G[from].push_back(to);
	G[to].push_back(from);
}
void dfs(int u,int fa,int fasfa){
	b[u]=true;
	if(fasfa)ans=(ans+d[u]*d[fasfa]%10007)%10007;
	int sum=0,no1=0,no2=0;
	for(int i=0;i<G[u].size();++i)
	if(!b[G[u][i]]){
		if(d[G[u][i]]>no1)no2=no1,no1=d[G[u][i]];else
		if(d[G[u][i]]>no2)no2=d[G[u][i]];
		ans=(ans+d[G[u][i]]*sum%10007)%10007;
		sum=(sum+d[G[u][i]])%10007;
	}
	if(no1*no2>Max)Max=no1*no2;
	if(d[u]*d[fasfa]>Max)Max=d[u]*d[fasfa];
	for(int i=0;i<G[u].size();++i)
	if(!b[G[u][i]]){
		b[G[u][i]]=true;
		dfs(G[u][i],u,fa);
	}
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<n;++i){
		int x,y;
		scanf("%d%d",&x,&y);
		addedge(x,y);
	}
	for(int i=1;i<=n;++i)scanf("%d",&d[i]);
	dfs(1,0,0);
	printf("%d %d\n",Max,ans*2%10007);
	return 0;
}