树-P1351 联合权值

题目链接：
https://www.luogu.org/problemnew/show/P1351
看题解，还是一知半解，先纪录一下，后续再搞。

/*
先看求联合权值的和。稍稍对题意进行变形，即为：对每一个点，找到与它相连的所有点，那些点之间两两权值相乘，最后把所有乘积相加（并取余，别忘了qwq）。
我们想象：每连上一条边，便进行一次“操作”。定义一个数组S，用S[i]记录第i个点周围已经连上的点的权值之和。连上一条边时，比方说从a点到b点，对这两个点分别进行操作，拿a点举例：
首先将b点的权值乘以已经连上a点的所有点的权值之和（就是S[a]），加到总结果sum上；然后把b点的权值加到S[a]上。
b点的操作同理。这样的话，从第一条边连到最后一条边，sum累积了一半的联合权值总和，x2便是正确答案。
（稍稍想想大概就是这么回事吧...）
再看求最大值。类似地，定义一个数组M，用M[i]记录第i个点周围已经连上的点的权值的最大值。类似于上面的操作，每连上一条边，分别更新相应的M[a]、M[b]，以及全局联合权值的最大值ans。
最后...别忘开long long否则70分orz
*/

#include<iostream>
#define N 200010
#define mod 10007

using namespace std;
long long w[N],A[N],B[N],S[N],M[N],sum,ans;
int n;
int main()
{
    cin>>n;
    for(int i=1;i<n;i++)   cin>>A[i]>>B[i];
    for(int i=1;i<=n;i++)   cin>>w[i];
    for(int i=1;i<n;i++)
    {
        int ta=A[i],tb=B[i];
        ans=max(ans,M[ta]*w[tb]);
        M[ta]=max(M[ta],w[tb]);//M[i]记录第i个点周围已经连上的点的权值的最大值
        sum=(sum+S[ta]*w[tb])%mod;
        S[ta]+=w[tb];//S[i]记录第i个点周围已经连上的点的权值之和
        
        ans=max(ans,M[tb]*w[ta]);
        M[tb]=max(M[tb],w[ta]);
        sum=(sum+S[tb]*w[ta])%mod;
        S[tb]+=w[ta];
    }
    cout<<ans<<" "<<(sum*2)%mod;
    return 0;
}