因为边权为1,所以说,对于一个点,所有与这个点直接连接的点间的距离均为2,对于一个点,设它直接连接的点为 a1,a2,a3 …… 那所求的结果为 a1*(a2 + a3 + …) + a2*(a1 + a3 + …)+a3*(a1 + a2 + …) + … 记sum = Σai , 上面的式子就是 Σai * (sum - ai),对每个点计算这个值即可。
#include<iostream>
#include<cstdio>
#include<cstring>
#define mod 10007
using namespace std;
const int size = 2100010;
int u[size],v[size];
int w[size];
int head[size],next[size],l[size];
int tot = 1;
void build(int f,int t)
{
l[tot] = t;
next[tot] = head[f];
head[f] = tot ++;
}
int main()
{
int n;
cin>>n;
for(int i = 1 ; i < n ; i ++)
{
scanf("%d%d",&u[i],&v[i]);
build(u[i],v[i]);
build(v[i],u[i]);
}
int ans = 0, maxn = 0;
for(int i = 1 ; i <= n ; i ++)
scanf("%d",&w[i]);
for(int i = 1 ; i <= n ; i ++)
{
int max1 = 0 , max2 = 0;
int sum = 0;
for(int j = head[i] ; j ; j = next[j])
{
int t = l[j];
sum += w[t];
if(w[t] >= max1)
{
max2 = max1;
max1 = w[t];
}
if(w[t] < max1 && w[t] > max2)
max2 = w[t];
}
maxn = max(maxn,max1*max2);
for(int j = head[i] ; j ; j = next[j])
ans += (((sum - w[l[j]])%mod) * (w[l[j]]%mod)) % mod;
}
cout<<maxn<<" "<<ans % mod;
return 0;
}
传送门 : codevs3728