又水了一道题，结果题目没看清以为是二叉树结果跪了好久。。。

Posted 2020-10-11 cc123321

题目描述

无向连通图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 

中间有个很妙的cnt，请自己体会。。

#include<cstdio>
#include<algorithm>
#include<vector>
#define ll long long
#define maxn 2000200
#define mod 10007 
using namespace std;
ll tot,n,m,u,v,ma,ans,dian[maxn],f[maxn];

vector <ll> s[maxn];

inline void dfs(ll pos)
{
    ll m1=0,m2=0,cnt=0;
    if(f[f[pos]])
    {
        ans += dian[pos] * dian[f[f[pos]]];
        ma = max(ma , dian[pos] * dian[f[f[pos]]]);
    }
    for(int i=0;i<s[pos].size();i++)
    {
        ll z = s[pos][i];
        if(z == f[pos]) continue;
        f[z] = pos;
        dfs(z);
        ans += cnt * dian[z];
        cnt += dian[z];
        ans %= mod;
        cnt %= mod;
        m2 = max(m2 , dian[z]);
        if(m2 > m1) swap(m1 , m2);
    }
    ma = max(ma , m1 * m2);
}

int main(){
    scanf("%lld",&n);

    for(ll i=1;i<n;i++)
    {
        scanf("%lld%lld",&u,&v);
        s[u].push_back(v);
        s[v].push_back(u);
    }

    for(ll i=1;i<=n;i++)
        scanf("%lld",&dian[i]);

    dfs(1);
    printf("%lld %lld",ma,ans * 2 % mod);
}