[noip2014]联合权值 题解

要距离为2，画一画图可以显然得到，就是每个点相连的每个点之间两两满足条件。

现在问题就成了，已知n个数的值,a1,a2,a3……an,求a1*a2+a1*a3+……a1*an+a2*a1+a2*a3…………。

然后稍微用一下数学公式，我们发现答案就是和的平方减去平方的和。

最大值维护一个最大值和次大值就好了。

#include<bits/stdc++.h> 
#define N 200000 
using namespace std;
int n,x,y,w[N+5];
int first[N+5],nxt[2*N+5],to[2*N+5],siz;
long long sum1,sum2,mx,ans,tmp1,tmp2;
inline void add(int x,int y) 
{
    nxt[siz]=first[x];
    first[x]=siz;
    to[siz]=y;
    siz++;
}
int main()
{
    freopen("in.txt","r",stdin);
    scanf("%d",&n);
    memset(first,-1,sizeof(first));
    for(int i=1;i<n;i++)scanf("%d%d",&x,&y),add(x,y),add(y,x);
    for(int i=1;i<=n;i++)scanf("%d",&w[i]);
    for(int i=1;i<=n;i++)
    {
        sum1=sum2=tmp1=tmp2=0;
        for(int j=first[i];j!=-1;j=nxt[j])
            sum1+=w[to[j]],sum2+=w[to[j]]*w[to[j]];
        for(int j=first[i];j!=-1;j=nxt[j])
            if(w[to[j]]>tmp1)tmp2=tmp1,tmp1=w[to[j]];
                else if(w[to[j]]>tmp2)tmp2=w[to[j]];
        mx=max(mx,tmp1*tmp2);
        ans=(ans+(sum1*sum1%10007-sum2))%10007;
    }
    cout<<mx<<" "<<(ans+10007)%10007;
    return 0;
}