【NOIP 2014 Day1 T2】联合权值（......加法结合律QAQ）

【问题描述】
无向连通图G有n个点，n-1条边。点从1到n依次编号，编号为i的点的权值为Wi ，每条边的长度均为1。图上两点(u, v)的距离定义为u点到v点的最短距离。对于图G上的点对(u, v)，若它们的距离为2，则它们之间会产生Wu×Wv的联合权值。 请问图G上所有可产生联合权值的有序点对中，联合权值最大的是多少？所有联合权值之和是多少？

【输入】
输入文件名为link.in。 第一行包含1个整数n。 接下来n-1行，每行包含2个用空格隔开的正整数u、v，表示编号为u和编号为v的点之间有边相连。
最后1行，包含n个正整数，每两个正整数之间用一个空格隔开，其中第i个整数表示图G上编号为i的点的权值为Wi。

【输出】
输出文件名为link.out。 输出共1行，包含2个整数，之间用一个空格隔开，依次为图G上联合权值的最大值和所有联合权值之和。由于所有联合权值之和可能很大，输出它时要对10007取余。

【输入输出样例】
样例输入 Sample Input
5
1 2
2 3
3 4
4 5
1 5 2 3 10

样例输出 Sample Output
20 74

【样例说明】

数据范围及提示 Data Size & Hint
对于30%的数据，1<≤100；
对于60%的数据，1<≤2000；
对于100%的数据，1<≤200,000，0< Wi ≤10,000

题解：让我们先考虑一下最暴力的方法，枚举每个点找跟它距离为2的点计算联合权值，这样显然复杂度很高。我们可以想一下两个点距离为2是什么情况，即这两个点连接着一个相同的点。这样我们就可以枚举中间点，找与它连接的点两两求联合权值，但是这样也不够优。这里有一个简单的优化，需要用到乘法结合律（没错，就是乘法结合律╮(╯▽╰)╭）。我们考虑一下枚举中间点来计算联合权值是什么情况，对于每个点都会跟其他与中间点相连的点产生一个联合权值，这样他们产生的联合权值和就是，w[i]*(sum-w[i])，sum是与中间点相连的点的权值和，这个式子应该不难理解233，这样这道题就能AC了。

代码如下：

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int MAXN=200100;
int first[MAXN],nxt[MAXN<<1];
int w[MAXN];
int n,sum=0,tot,maxx=0;
struct edge
{
    int from,to;
}es[MAXN<<1];
void init()
{
    memset(first,-1,sizeof(first));
    tot=0;
}
void build(int f,int t)
{
    es[++tot]=(edge){f,t};
    nxt[tot]=first[f];
    first[f]=tot;
}
int main()
{
    scanf("%d",&n);
    init();
    for(int i=1;i<n;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        build(u,v);
        build(v,u);
    }
    for(int i=1;i<=n;i++)
        scanf("%d",&w[i]);
    for(int i=1;i<=n;i++)
    {
        int s=0,max1=0,max2=0;
        for(int j=first[i];j!=-1;j=nxt[j])
        {
            int v=es[j].to;
            s+=w[v];
            s%=10007;//计算与点i相连的所有点的权值和 
        }
        for(int j=first[i];j!=-1;j=nxt[j])
        {
            int v=es[j].to;
            if(w[v]>=s) sum+=(s-w[v]+10007)*w[v];
            else sum+=(s-w[v])*w[v];
            sum%=10007;//计算联合权值和 
            if(w[v]>max1)//存一下与这个点相连的点的权值最大和次大值 
            {
                max2=max1;
                max1=w[v];
            }
            else if(w[v]>max2)
                max2=w[v];

        }
        if(max1*max2>maxx)//求联合权值最大值 
            maxx=max1*max2;
    }
    printf("%d %d\n",maxx,sum%10007);
    return 0;
}
/*
5
1 2
2 3
3 4
4 5
1 5 2 3 10
*/