【NOIP2014】联合取值

联合取值

题目描述

无向连通图G 有n个点，n - 1 条边。点从1 到n依次编号，编号为 i 的点的权值为W i ，每条边的长度均为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 的点的权值为W i。

输出格式：
 

输出文件名为link .out。

输出共1 行，包含2个整数，之间用一个空格隔开，依次为图G上联合权值的最大值

和所有联合权值之和。由于所有联合权值之和可能很大，输出它时要对10007取余。

输入输出样例

输入样例#1： 

5  

1 2  

2 3

3 4  

4 5  

1 5 2 3 10 

输出样例#1： 

20 74

本例输入的图如上所示，距离为2的有序点对有( 1,3)、( 2,4) 、( 3,1) 、(3,5)、( 4,2) 、( 5,3) 。

其联合权值分别为2 、15、2、20、15、20。其中最大的是20，总和为74。

【数据说明】

对于30% 的数据，1 < n≤ 100；

对于60% 的数据，1 < n≤ 2000；

对于100%的数据，1 < n≤ 200 , 000，0 < wi≤ 10, 000。

 

一开始来个暴力优化期盼能过结果就多了10分70……

距离为2，也就是每次都会有个转接点，于是我们就枚举这个转接点。对于每个转接点，有两个点ab想接的话，就是2ab，三个点的话就是2ab,2ac,2bc。但是我们发现直接硬搞有些麻烦。所以我们来转化一下，2ab=(a+b)^2-(a^2+b^2)，2ab+2ac+2ac=(a+b)^2+2c(a+b)^2+c^2-(a^2+b^2+c^2)=(a+b+c)^2-(a^2+b^2+c^2)。就ac了

 

 

#include<iostream>
#include<cstdio>
#include<cstring>
#define N  405000
#define P 10007
using namespace std;

int head[N],next[N],to[N],a[N];
int tot=0,n;
long long ans,maxx;

void add_edge(int x,int y)
{
    to[++tot]=y;
    next[tot]=head[x];
    head[x]=tot;
}

int main()
{
    ans=0;maxx=0;
    scanf("%d",&n);
    for (int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        add_edge(x,y);
        add_edge(y,x);
    }
    for (int i=1;i<=n;i++) scanf("%d",&a[i]);
    for (int j=1;j<=n;j++)
    {
        long long x=j,sum1=0,sum2=0,max1=0,max2=0;
        for (int i=head[x];i;i=next[i])
        {
            int y=to[i];
            if (a[y] > max1) max2=max1,max1=a[y];
            else if (a[y] > max2) max2=a[y];
            sum1=sum1+a[y]*a[y];
            sum2+=a[y];
        }
        if (max1*max2 > maxx) maxx=max1*max2;
        ans=(ans+sum2*sum2-sum1)%P;
    }
    printf("%lld %lld\n",maxx,ans);
    return 0;
}