题目:给定一个二叉树,找到最长的路径,这个路径中的每个节点具有相同值。 这条路径可以经过也可以不经过根节点。
注意:两个节点之间的路径长度由它们之间的边数表示。
示例 1:
输入:
5
/ \
4 5
/ \ \
1 1 5
输出:
2
此题同最长路径和的解法, 在递归遍历的过程中不会直接返回结果,而是在不断遍历的过程中比较得到的结果,最终求得最值。
如下:
int ans = 0;
public int longestUnivaluePath(TreeNode root) {
oneSideLongestUnivaluePath(root);
return ans;
}
// 先定义好递归的含义,该方法求的是单边最长的同值路径
public int oneSideLongestUnivaluePath(TreeNode root) {
if (root == null) {
return 0;
}
// 递归找左子树的最长同值路径
int left = oneSideLongestUnivaluePath(root.left);
// 同理
int right = oneSideLongestUnivaluePath(root.right);
int leftArrow = 0;
// 如果左节点与根值相同,那么同值路径加1
if (root.left != null && root.val == root.left.val) {
leftArrow = (left + 1);
}
int rightArrow = 0;
if (root.right != null && root.val == root.right.val) {
rightArrow = (right + 1);
}
// 不断的比较单边路径长度,找到最大的。
ans = Math.max(leftArrow + rightArrow, ans);
return Math.max(leftArrow, rightArrow); // 注意,方法的定义是单边的最长同值路径。
}
参考leetcode124题 【二叉树中的最大路径和】,此题与本题有相似之处。