树木边缘切割

Question

通过移除其中一个边，树可以分成两个不同的树。 给定一个树，其中N节点由[0, N-1]范围内的整数唯一标识，我需要编写一个函数来查找需要从树中删除的边，这样所有的总和之间的差异结果树中的节点ID被最小化。

该函数应该打印它找到的标准输出（stdout）的最小差异。

该函数将收到以下参数：

parent是一个整数数组，具有以下含义： parent[i] =节点i的父节点（更具体地说是其ID）

如果我没有父（我是树的根）， parent[i] = -1

数据限制

树中的最大节点数为50,000

效率限制

该功能有望在不到2秒的时间内打印出结果

例

 Input parent: [1, 4, 4, 2, -1, 2, 2]

 aka :         4
              / \
             1   2
            /  / | \
           0  3  5  6   

 Output: 9

说明：我们删除节点2和6之间的边缘。

Answer 1

1. 对于每个节点，计算其子节点的总和，然后将此值添加到自身，并将其存储在节点中。 我们将此值称为S_n ，其中n是节点。 （可以通过递归和后序遍历轻松完成）

2. 找到S_n与S_root/2具有最小差异的节点n 。 n与其父级之间的边缘是我们想要的边缘。 （在最坏的情况下，这需要线性时间。）