方法:递归

思路

我们可以将任何路径(具有相同值的节点)看作是最多两个从其根延伸出的箭头。

具体地说,路径的根将是唯一节点,因此该节点的父节点不会出现在该路径中,而箭头将是根在该路径中只有一个子节点的路径。

然后,对于每个节点,我们想知道向左延伸的最长箭头和向右延伸的最长箭头是什么?我们可以用递归来解决这个问题。

算法

arrow_length(node) 为从节点 node 延伸出的最长箭头的长度。如果 node.Left 存在且与节点 node 具有相同的值,则该值就会是 1 + arrow_length(node.left)。在 node.right 存在的情况下也是一样。

当我们计算箭头长度时,候选答案将是该节点在两个方向上的箭头之和。我们将这些候选答案记录下来,并返回最佳答案。

复杂度分析

  • 时间复杂度:,其中 是树中节点数。我们处理每个节点一次。

  • 空间复杂度:,其中 是树的高度。我们的递归调用栈可以达到 层的深度。