[NOIP2014]联合权值 题解

题目大意：

　　有一棵树，求距离为2的点权的乘积的和以及最大值。

思路：

　　枚举每一个点，则与其相邻的点互为距离为2的点。该部分的最大值为点权最大的两个点的积，和为点的权值和的平方减去每个点的平方，这样每条边都被跑了两次，所以复杂度为O(n)。

　　用邻接表存储要开双倍数组（无向），当然像cyk大神一样直接跑边就不用考虑这个了。

代码：

　　邻接表：

#include<cstdio>
const int mo=10007,M=200008;
int cnt,x,y,n,i,ans,tot,w[M],v[M<<1],last[M<<1],head[M<<1];

void add(int x,int y) { v[++cnt]=y,last[cnt]=head[x],head[x]=cnt; }

int main()
{
    scanf("%d",&n);
    for (i=1;i<n;i++) scanf("%d%d",&x,&y),add(x,y),add(y,x);
    for (i=1;i<=n;i++) scanf("%d",&w[i]);
    for (i=1;i<=n;i++)
    {
        int sum=0,max1=0,max2=0,j,o;
        for (j=head[i];j;j=last[j])
        {
            o=w[v[j]];
            sum=(sum+o)%mo;
            if (o>max1) max2=max1,max1=o;
            else if (o>max2) max2=o;
            tot=(tot-o*o)%mo;
        }
        tot=(tot+sum*sum)%mo;
        sum=max1*max2;
        if (sum>ans) ans=sum;
    }
    printf("%d %d\n",ans,tot);
    return 0;
}

　　%%%cyk大神：

#include<iostream>
using namespace std;
int i,n,maxx,ans,x[200005],y[200005],w[200005];
int am[200005],am2[200005],s[200005],qs[200005];
int main()
{
    cin>>n;
    for (i=1;i<=n-1;i++) cin>>x[i]>>y[i];
    for (i=1;i<=n;i++) cin>>w[i];
    for (i=1;i<=n-1;i++)
      {
          s[x[i]]=(s[x[i]]+w[y[i]])%10007;
          qs[x[i]]=(qs[x[i]]+w[y[i]]*w[y[i]])%10007;
          s[y[i]]=(s[y[i]]+w[x[i]])%10007;
          qs[y[i]]=(qs[y[i]]+w[x[i]]*w[x[i]])%10007;
          if (w[y[i]]>am[x[i]]) {am2[x[i]]=am[x[i]];am[x[i]]=w[y[i]];}
          else if (w[y[i]]>am2[x[i]]) am2[x[i]]=w[y[i]];
          if (w[x[i]]>am[y[i]]) {am2[y[i]]=am[y[i]];am[y[i]]=w[x[i]];}
          else if (w[x[i]]>am2[y[i]]) am2[y[i]]=w[x[i]];
      }
    maxx=0;
    for (i=1;i<=n;i++) maxx=max(maxx,am[i]*am2[i]);
    ans=0;
    for (i=1;i<=n;i++) ans=(ans+s[i]*s[i]-qs[i])%10007;
    cout<<maxx<<' '<<ans;
}