洛谷P1351 联合权值（NOIP 2014）

题目描述 https://www.luogu.org/problemnew/show/P1351
首先说一个公式：一些数和的平方减去这些数平方的和等于这些数两两相乘的和，本题会用到。
找相距为2的两个点，其实就是一个点相连的两个不同的点，这两个点的距离等于2。
我们要求的是最大的权值还有所有权值的和，所以我们要记一下与每个点相连的这些点的和以及平方和，为求和做准备。，还要记下这些点值的最大值和次大值，为后面求最大值作准备。
记下这些，这道题基本上就完成了。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=2e5+5,modd=10007;
int h[N],to[2*N],next[2*N],tot,ans,sum;
int m1[N],m2[N],s1[N],s2[N],w[N],n;
void add(int x,int y)
{
	to[++tot]=y;
	next[tot]=h[x];
	h[x]=tot;
}
void work()
{
	for(int i=1;i<=n;i++)
	  for(int j=h[i];j;j=next[j])
	  {
	  	 int y=to[j];
		 s1[i]=(s1[i]+w[y])%modd,s2[i]=(s2[i]+w[y]*w[y])%modd;
		 if(w[y]>m1[i]) m2[i]=m1[i],m1[i]=w[y];
		 else if(w[y]>m2[i]) m2[i]=w[y];
	  }
}
int main() 
{ 
   scanf("%d",&n);
   for(int i=1;i<=n-1;i++)
   {
   	  int x,y;
   	  scanf("%d%d",&x,&y);
   	  add(x,y),add(y,x); 
   }
   for(int i=1;i<=n;i++) scanf("%d",&w[i]);
   work();
   for(int i=1;i<=n;i++)
   {
   	   sum=(sum+(s1[i]*s1[i]-s2[i]+modd)%modd)%modd;
   	   ans=max(ans,m1[i]*m2[i]);
   }
   printf("%d %d",ans,sum);
 	
  return 0;	
}