洛谷P1351 [NOIP2014 提高组] 联合权值【树形DP】【绿】

Date：2022.04.04
题目描述
无向连通图 G 有 n 个点，n-1 条边。点从 1 到 n 依次编号,编号为 i 的点的权值为 Wi，每条边的长度均为 1。图上两点 (u, v) 的距离定义为 u 点到 v 点的最短距离。对于图 G 上的点对 (u, v)，若它们的距离为 2，则它们之间会产生Wv×Wu的联合权值。
请问图 G 上所有可产生联合权值的有序点对中，联合权值最大的是多少？所有联合权值之和是多少？
输入格式
第一行包含 1 个整数 n。
接下来 n−1 行,每行包含 2 个用空格隔开的正整数 u,v，表示编号为 u 和编号为 v 的点之间有边相连。
最后 1 行,包含 n 个正整数，每两个正整数之间用一个空格隔开，其中第 ii 个整数表示图 G 上编号为 i 的点的权值为 Wi。
输出格式
输出共 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≤200000,0<Wi≤10000。
保证一定存在可产生联合权值的有序点对。

思路：首先未说有向边，我们认定为双向边。无相连通图，且有$n-1$条边，那一定是个树！
我们注意到每个人只会和其子节点的子节点有交集，树形dp讨论下即可。然而不要漏掉它的兄弟结点和它也是一个合法组合（其实它父节点的父节点和它也是一个组合，并且在求和时不能忽略，但由于我们前面讨论了，再反向一下就行了，不必再讨论，因此我们只需要每次向下递归，根本不需要考虑向上的问题）。那我们就枚举每个节点的每个兄弟结点？显然t飞了。我们观察得到，每个节点的所有兄弟结点都有一个性质：和其父节点的距离为1。因此我们可以初始化出来以每个点$u$为根节点时：
①所有与$u$距离为1的点的权值和（包括其父节点$fa[u]$）。这是为了求$u$的子节点$j_i$与$u$的所有除了$j_i$外、且距离是1的点乘积之和，因为这些距离都是2。
②所有与$u$距离为1的点中最大权值点和次大权值点及它们的结点编号。这是为了更新最大乘积，但每次讨论一个结点 j i j_i j