题目: 给定一个二叉树,找到最长的路径,这个路径中的每个节点具有相同值。 这条路径可以经过也可以不经过根节点。注意:两个节点之间的路径长度由它们之间的边数表示。
示例:
1.
2.
思路分析:
一、最长同值路径分为两种情况
1.1 以root为起点,经过左子树或右子树,如上图(2)所示。
1.2 不是以root为起点,root是中间点,如上图(1)所示。
二、但是如果我们设计的递归函数helper,递归计算以每一个节点为起点的最长同值路径(只计算了情况1的结果),如下所示:
ps:橘色数字代表以该结点为起点的最长同值路径。(故以5的maxLength是1而不是2,因为2是以5为中间结点的最长同值路径),但是我们完全可以借助只计算了情况1的递归函数helper来计算包含情况2的结果。
算法复杂度分析:
1.时间复杂度:O(N),对所有结点遍历一次且仅一次。
2.空间复杂度:O(N),递归需要调用辅助栈空间,且只定义了几个变量。
Code:
class Solution {
int ans = 0;
public int longestUnivaluePath(TreeNode root) {
helper(root);
return ans;
}
public int helper(TreeNode root) {
if (root == null) {
return 0;
}
int maxLength = 0;
int leftLength = helper(root.left);
int rightLength = helper(root.right);
if (root.left != null && root.left.val == root.val && root.right != null && root.right.val == root.val) {
ans = Math.max(ans, leftLength + rightLength + 2);
}
if (root.left != null && root.val == root.left.val) {
maxLength = leftLength + 1;
}
if (root.right != null && root.right.val == root.val) {
maxLength = Math.max(maxLength, rightLength + 1);
}
ans = Math.max(ans, maxLength);
return maxLength;
}
}