每日一题 P1351 [NOIP2014 提高组] 联合权值 树上问题

思路很简单，每个点能构成联合权值的，只有他的兄弟和他的爷爷。因此只要跑一遍dfs就可以计算出来。
数组的含义 ：son一个节点的儿子的和 smax 一个节点的儿子的最大值。
因为算联合权值的时候两个节点(a,b)互换一下位置(b,a)也应该算在答案内，所以最后ans我们需要给他*2。

#include <iostream>
#include <cstring>
#define MAXN 500005
using namespace std;
const int MOD=10007;
struct EDGE
{
    int to,next;
} edge[MAXN];
int head[MAXN],ptr;
int val[MAXN],son[MAXN],smax[MAXN];
void add_edge(int u,int v)
{
    edge[++ptr].to=v;
    edge[ptr].next=head[u];
    head[u]=ptr;
}
void add(int u,int v)
{
    add_edge(u,v);
    add_edge(v,u);
}
int N,ans,ans2;
void dfs(int now,int fa)
{
    for(int p=head[now];p;p=edge[p].next)
    {
        int to=edge[p].to;
        if(to==fa)continue;
        dfs(to,now);
        ans+=son[now]*val[to],ans%=MOD;
        son[now]+=val[to],son[now]%=MOD;
        ans2=max(ans2,max(val[fa],smax[now])*val[to]);
        smax[now]=max(smax[now],val[to]);
    }
    ans+=val[fa]*son[now],ans%=MOD;
}
int main()
{
    cin>>N;
    for(int i=1;i<N;i++)
    {
        int a,b;cin>>a>>b;
        add(a,b);
    }
    for(int i=1;i<=N;i++)
        cin>>val[i];
    dfs(1,0);
    cout<<ans2<<' '<<ans*2%MOD;
}