NOIP 2014提高组 联合权值

因为边权为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